Searched defs:shape_tensor (Results 1 - 6 of 6) sorted by relevance
/external/tensorflow/tensorflow/python/ops/linalg/ |
H A D | linear_operator_util.py | 292 def shape_tensor(shape, name=None): function
|
H A D | linear_operator.py | 288 def shape_tensor(self, name="shape_tensor"): member in class:LinearOperator 306 self._cached_shape_tensor = linear_operator_util.shape_tensor( 344 self._cached_batch_shape_tensor = linear_operator_util.shape_tensor( 347 self._cached_batch_shape_tensor = self.shape_tensor()[:-2] 387 self._cached_tensor_rank_tensor = array_ops.size(self.shape_tensor()) 425 self._cached_domain_dimension_tensor = self.shape_tensor()[-1] 463 self._cached_range_dimension_tensor = self.shape_tensor()[-2]
|
/external/tensorflow/tensorflow/contrib/boosted_trees/lib/testutil/ |
H A D | batch_features_testutil.cc | 74 auto shape_tensor = Tensor(tensorflow::DT_INT64, {2}); local 75 tensorflow::test::FillValues<int64>(&shape_tensor, {batch_size, 1}); 76 sparse_float_feature_shapes_list.push_back(shape_tensor);
|
/external/tensorflow/tensorflow/contrib/libsvm/kernels/ |
H A D | decode_libsvm_op.cc | 125 Tensor* shape_tensor; variable 128 &shape_tensor)); 129 auto shape = shape_tensor->flat<int64>();
|
/external/tensorflow/tensorflow/core/kernels/ |
H A D | parameterized_truncated_normal_op.cc | 254 const Tensor& shape_tensor = ctx->input(0); variable 261 ctx, TensorShapeUtils::IsVector(shape_tensor.shape()), 263 shape_tensor.shape().DebugString())); 264 int32 num_batches = shape_tensor.flat<int32>()(0); 267 const int32 num_dims = shape_tensor.dim_size(0); 269 samples_per_batch *= shape_tensor.flat<int32>()(i); 274 auto shape_vec = shape_tensor.flat<int32>();
|
H A D | sparse_conditional_accumulator.h | 442 Tensor* shape_tensor; local 444 ctx, ctx->allocate_output(2, {accum_val_dims}, &shape_tensor)); 449 shape_tensor->flat<int64>()(0) = 452 shape_tensor->flat<int64>()(i) = accum_val_->dim_size(i);
|
Completed in 720 milliseconds