| /external/tensorflow/tensorflow/python/ops/linalg/ |
| linear_operator_util.py | 109 def shape_tensor(shape, name=None): function
|
| linear_operator.py | 280 def shape_tensor(self, name="shape_tensor"): member in class:LinearOperator
296 return linear_operator_util.shape_tensor(self.shape.as_list())
331 return linear_operator_util.shape_tensor(
334 return self.shape_tensor()[:-2]
371 return array_ops.size(self.shape_tensor())
410 return self.shape_tensor()[-1]
449 return 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/core/kernels/ |
| broadcast_to_op.cc | 44 const Tensor& shape_tensor = ctx->input(1); variable
48 ctx->op_kernel().MakeShape(shape_tensor, &output_shape));
|
| parameterized_truncated_normal_op.cc | 329 const Tensor& shape_tensor = ctx->input(0); variable
336 ctx, TensorShapeUtils::IsVector(shape_tensor.shape()),
338 shape_tensor.shape().DebugString()));
339 int32 num_batches = shape_tensor.flat<int32>()(0);
342 const int32 num_dims = shape_tensor.dim_size(0);
344 samples_per_batch *= shape_tensor.flat<int32>()(i);
349 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);
|
| /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/framework/ |
| common_shape_fns.cc | 1639 const Tensor* shape_tensor = c->input_tensor(2); local
[all...] |
