| /external/tensorflow/tensorflow/python/ops/linalg/ |
| linear_operator_util.py | 292 def shape_tensor(shape, name=None): function
|
| 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]
[all...] |
| /external/tensorflow/tensorflow/contrib/boosted_trees/lib/testutil/ |
| 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/ |
| decode_libsvm_op.cc | 125 Tensor* shape_tensor; variable
128 &shape_tensor));
129 auto shape = shape_tensor->flat<int64>();
|
| /external/tensorflow/tensorflow/core/kernels/ |
| 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>();
|
| 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);
|
