| /external/tensorflow/tensorflow/core/profiler/internal/ |
| D | tfprof_node.cc | 266 std::vector<int64> shape_vec; in ShapeProtoToVec() local
269 shape_vec.push_back(1); in ShapeProtoToVec()
272 shape_vec.push_back(d.size()); in ShapeProtoToVec()
275 return shape_vec; in ShapeProtoToVec()
278 TensorShapeProto VecToShapeProto(const std::vector<int64>& shape_vec) { in VecToShapeProto() argument
280 if (shape_vec.empty()) { in VecToShapeProto()
284 for (const int64 s : shape_vec) { in VecToShapeProto()
|
| D | tfprof_show.cc | 227 std::vector<string> shape_vec; in FormatNode() local
230 shape_vec.push_back(strings::Printf("%d:unknown", s.first)); in FormatNode()
232 shape_vec.push_back(strings::Printf( in FormatNode()
236 info.push_back(str_util::Join(shape_vec, "|")); in FormatNode()
|
| D | tfprof_node.h | 41 TensorShapeProto VecToShapeProto(const std::vector<int64>& shape_vec);
709 std::vector<int64>& shape_vec = input_shapes[inp.first]; in input_shapes() local
723 shape_vec.assign(output_shape->second.begin(), in input_shapes()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | random_crop_op.cc | 47 auto shape_vec = shape_t.vec<int64>(); in Compute() local
48 const int32 target_height = shape_vec(0); in Compute()
49 const int32 target_width = shape_vec(1); in Compute()
|
| D | avgpooling_op.cc | 260 auto shape_vec = tensor_in_shape.vec<int32>(); in Compute() local
262 output_shape.AddDim(shape_vec(i)); in Compute()
415 auto shape_vec = tensor_in_shape.vec<int32>(); in Compute() local
417 output_shape.AddDim(shape_vec(i)); in Compute()
494 auto shape_vec = tensor_in_shape.vec<int32>(); in Compute() local
496 output_shape.AddDim(shape_vec(i)); in Compute()
|
| D | sparse_dense_binary_op_shared.cc | 83 const auto shape_vec = shape_t->vec<int64>(); in Compute() local
84 const auto lhs_dims = BCast::FromShape(TensorShape(shape_vec)); in Compute()
|
| D | sparse_reduce_op.cc | 174 const auto shape_vec = shape_t->vec<int64>(); in Compute() local
178 TensorShape(shape_vec), &sp)); in Compute()
200 output_strides[d + 1] * shape_vec(reduction.group_by_dims[d + 1]); in Compute()
|
| D | parameterized_truncated_normal_op.cc | 349 auto shape_vec = shape_tensor.flat<int32>(); in Compute() local
352 shape_vec.data(), shape_vec.size(), &tensor_shape)); in Compute()
|
| D | mkl_avgpooling_op.cc | 252 auto shape_vec = tensor_in_shape.vec<int32>(); in Compute() local
254 output_shape.AddDim(shape_vec(i)); in Compute()
589 auto shape_vec = orig_input_tensor.vec<int32>(); in Compute() local
592 orig_input_shape.AddDim(shape_vec(i)); in Compute()
|
| D | sparse_cross_op.cc | 531 auto shape_vec = (*shape_out)->vec<int64>(); in CreateOutputTensors() local
532 shape_vec(0) = batch_size; in CreateOutputTensors()
533 shape_vec(1) = max_cross_count; in CreateOutputTensors()
|
| D | remote_fused_graph_execute_utils.cc | 855 std::vector<TensorShape> shape_vec; in BuildClusterSubgraphDef() local
856 GetOutputTensorShapeType(*node_def, &dt_vec, &shape_vec).IgnoreError(); in BuildClusterSubgraphDef()
860 shape_vec.empty() ? TensorShape({}) : shape_vec.at(subgraph_input_port); in BuildClusterSubgraphDef()
1416 std::vector<TensorShape> shape_vec; in BuildNodeMapFromOpsDefinitions() local
1418 GetOutputTensorShapeType(node_def, &dt_vec, &shape_vec); in BuildNodeMapFromOpsDefinitions()
1420 shape_vec.clear(); in BuildNodeMapFromOpsDefinitions()
|
| D | pooling_ops_3d.cc | 505 auto shape_vec = tensor_in_shape.vec<int32>(); in Compute() local
507 output_shape.AddDim(shape_vec(i)); in Compute()
|
| /external/tensorflow/tensorflow/contrib/image/kernels/ |
| D | image_ops.cc | 95 auto shape_vec = shape_t.vec<int32>(); in Compute() local
96 out_height = shape_vec(0); in Compute()
97 out_width = shape_vec(1); in Compute()
|
| /external/tensorflow/tensorflow/contrib/layers/kernels/ |
| D | sparse_feature_cross_kernel.cc | 574 auto shape_vec = (*shape_out)->vec<int64>(); in CreateOutputTensors() local
575 shape_vec(0) = batch_size; in CreateOutputTensors()
576 shape_vec(1) = max_cross_count; in CreateOutputTensors()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | common_shape_fns.cc | 1642 auto shape_vec = shape_tensor->flat<int64>(); in SparseReduceShapeFn() local
1645 int64 ndims = shape_vec.size(); in SparseReduceShapeFn()
1656 dims.push_back(c->MakeDim(shape_vec(d))); in SparseReduceShapeFn()
1664 dims.push_back(c->MakeDim(shape_vec(d))); in SparseReduceShapeFn()
|