| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| sendrecv_ops.cc | 68 TensorShape tensor_shape; local
70 OP_REQUIRES_OK(ctx, ctx->GetAttr("shape", &tensor_shape));
72 OP_REQUIRES_OK(ctx, TensorShapeToXLAShape(dtype, tensor_shape, &shape_));
|
| /external/tensorflow/tensorflow/contrib/boosted_trees/lib/utils/ |
| batch_features.cc | 97 auto tensor_shape = TensorShape({shape_flat(0), shape_flat(1)}); local
101 tensor_shape, order_dims);
137 auto tensor_shape = TensorShape({shape_flat(0), shape_flat(1)}); local
141 tensor_shape, order_dims);
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| hlo_tfgraph_builder.cc | 23 #include "tensorflow/core/framework/tensor_shape.pb.h"
51 TensorShapeProto tensor_shape; local
54 tensor_shape.add_dim()->set_size(dim);
56 return tensor_shape;
|
| /external/tensorflow/tensorflow/contrib/util/ |
| convert_graphdef_memmapped_format_lib.cc | 24 #include "tensorflow/core/framework/tensor_shape.pb.h"
51 const TensorShapeProto tensor_shape = tensor_proto.tensor_shape(); local
97 *(attr_value.mutable_shape()) = tensor_shape;
|
| /external/tensorflow/tensorflow/core/kernels/ |
| parameterized_truncated_normal_op.cc | 34 #include "tensorflow/core/framework/tensor_shape.h"
275 TensorShape tensor_shape; variable
277 shape_vec.data(), shape_vec.size(), &tensor_shape));
279 OP_REQUIRES_OK(ctx, ctx->allocate_output(0, tensor_shape, &samples_tensor));
|
| load_and_remap_matrix_op.cc | 138 TensorShape tensor_shape; variable
140 old_tensor_name, &tensor_type, &tensor_shape));
148 context, tensor_shape.dims() == 2,
151 tensor_shape.DebugString(), " of invalid rank ",
152 tensor_shape.dims(), " instead of expected shape of rank 2.")));
158 OP_REQUIRES(context, num_cols_ == tensor_shape.dim_size(1),
161 tensor_shape.DebugString(),
163 tensor_shape.dim_size(1),
170 TensorSlice slice(tensor_shape.dims());
207 tensor_slice.SliceTensorShape(tensor_shape, &slice_shape))
[all...] |
| sparse_conditional_accumulator.h | 89 const Tensor* tensor_shape = std::get<2>(*tensor); local
95 if (shape_.dims() > tensor_shape->NumElements()) {
98 ", got ", tensor_shape->NumElements());
100 const auto tensor_shape_flat = tensor_shape->flat<int64>();
|
| random_op.cc | 29 #include "tensorflow/core/framework/tensor_shape.h"
184 TensorShape tensor_shape; local
185 TF_RETURN_IF_ERROR(ctx->op_kernel().MakeShape(shape, &tensor_shape));
186 return ctx->allocate_output(index, tensor_shape, output);
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| xla_op_kernel.cc | 350 TensorShape tensor_shape; local
352 XLAShapeToTensorShape(*shape.ValueOrDie(), &tensor_shape));
354 context_->allocate_output(index, tensor_shape, &output));
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| tensor_queue_dataset_op.cc | 532 const PartialTensorShape& tensor_shape = padded_shapes[i]; variable
535 IsGreaterEqualToOrCompatibleWith(tensor_shape,
540 " vs. ", tensor_shape.DebugString()));
|
| /external/tensorflow/tensorflow/c/ |
| c_api_function_test.cc | 1511 TF_Tensor* tensor_shape = Int32Tensor({37, 1}); local
[all...] |
| /external/tensorflow/tensorflow/contrib/mpi_collectives/kernels/ |
| mpi_ops.cc | 339 TensorShape tensor_shape = requests[0].tensor_shape(); local
345 TensorShape request_shape = requests[i].tensor_shape();
346 if (tensor_shape != request_shape) {
350 << tensor_shape.DebugString()
362 TensorShape tensor_shape = requests[0].tensor_shape(); local
364 if (tensor_shape.dims() == 0) {
374 TensorShape request_shape = requests[i].tensor_shape();
375 if (tensor_shape.dims() != request_shape.dims())
[all...] |
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| op_level_cost_estimator.cc | 21 #include "tensorflow/core/framework/tensor_shape.pb.h"
830 auto tensor_shape = local
832 for (const auto& dim : tensor_shape.dim()) {
[all...] |
| /external/tensorflow/tensorflow/contrib/lite/toco/ |
| export_tensorflow.cc | 34 #include "tensorflow/core/framework/tensor_shape.pb.h"
310 auto* tensor_shape = tensor->mutable_tensor_shape(); local
311 tensor_shape->add_dim()->set_size(shape.size());
[all...] |
| /external/tensorflow/tensorflow/python/eager/ |
| pywrap_tfe_src.cc | 473 const auto& tensor_shape = default_value.shape(); local
474 if (tensor_shape.unknown_rank()) {
477 const auto num_dims = tensor_shape.dim_size();
480 dims[i] = tensor_shape.dim(i).size();
[all...] |
