| /external/tensorflow/tensorflow/compiler/tf2tensorrt/utils/ |
| D | trt_shape_optimization_profiles_test.cc | 151 std::vector<TensorShape> shape_vec = DimVecToShapeVec(dim_vec); in TEST_F() local
170 std::vector<TensorShape> shape_vec = DimVecToShapeVec(dim_vec); in TEST_F() local
192 std::vector<TensorShape> shape_vec = DimVecToShapeVec(dimvec); in TEST_F() local
|
| D | trt_shape_optimization_profiles.cc | 30 std::vector<nvinfer1::Dims> GetDimVec(std::vector<TensorShapeType> shape_vec) { in GetDimVec()
62 for (auto& shape_vec : input_shapes_) { in ImplicitBatchModeCompatibleStrategy() local
74 for (auto& shape_vec : input_shapes_) { in OptimalStrategy() local
|
| /external/tensorflow/tensorflow/core/profiler/internal/ |
| D | tfprof_node.cc | 261 std::vector<int64> shape_vec; in ShapeProtoToVec() local
273 TensorShapeProto VecToShapeProto(const std::vector<int64>& shape_vec) { in VecToShapeProto()
|
| D | tfprof_show.cc | 227 std::vector<string> shape_vec; in FormatNode() local
|
| D | tfprof_node.h | 708 std::vector<int64>& shape_vec = input_shapes[inp.first]; in input_shapes() local
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | avgpooling_op.cc | 294 auto shape_vec = tensor_in_shape.vec<int32>(); in Compute() local
452 auto shape_vec = tensor_in_shape.vec<int32>(); in Compute() local
538 auto shape_vec = tensor_in_shape.vec<int32>(); in Compute() local
|
| D | sparse_dense_binary_op_shared.cc | 83 const auto shape_vec = shape_t->vec<int64>(); in Compute() local
|
| D | sparse_reduce_op.cc | 174 const auto shape_vec = shape_t->vec<int64>(); in Compute() local
|
| D | parameterized_truncated_normal_op.cc | 640 auto shape_vec = shape_tensor.flat<int32>(); in Compute() local
|
| D | pooling_ops_3d.cc | 501 auto shape_vec = tensor_in_shape.vec<int32>(); in Compute() local
|
| D | remote_fused_graph_execute_utils.cc | 857 std::vector<TensorShape> shape_vec; in BuildClusterSubgraphDef() local
1418 std::vector<TensorShape> shape_vec; in BuildNodeMapFromOpsDefinitions() local
|
| D | sparse_cross_op.cc | 684 auto shape_vec = (*shape_out)->vec<int64>(); in CreateOutputTensors() local
|
| /external/tensorflow/tensorflow/core/kernels/image/ |
| D | random_crop_op.cc | 47 auto shape_vec = shape_t.vec<int64>(); in Compute() local
|
| D | image_ops.cc | 77 auto shape_vec = shape_t.vec<int32>(); in DoImageProjectiveTransformOp() local
|
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_avgpooling_op.cc | 194 auto shape_vec = orig_input_tensor.vec<int32>(); in Compute() local
|
| /external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/ |
| D | legalize_tf.cc | 1007 llvm::SmallVector<int64_t, 4> shape_vec; in matchAndRewrite() local
|
| D | legalize_common.cc | 2330 SmallVector<int64_t, 4> shape_vec(input_shape.begin(), input_shape.end()); in convertReduceOpCommon() local
|
| D | legalize_tfl.cc | 1247 llvm::SmallVector<int64_t, 4> shape_vec; in matchAndRewrite() local
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | common_shape_fns.cc | 2422 auto shape_vec = shape_tensor->flat<int64>(); in SparseReduceShapeFn() local
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_nodes_test.cc | 1323 std::vector<TensorShape>* shape_vec) { in GetShapeFromDataVec()
|