| /external/tensorflow/tensorflow/compiler/tf2tensorrt/utils/ |
| D | trt_shape_optimization_profiles_test.cc | 113 std::vector<TensorShape> shape_vec = DimVecToShapeVec(dimvec); in CheckProfile() local
180 std::vector<TensorShape> shape_vec = DimVecToShapeVec(dim_vec); in TEST_P() local
201 std::vector<TensorShape> shape_vec = DimVecToShapeVec(dim_vec, true); in TEST_P() local
|
| D | trt_shape_optimization_profiles.cc | 34 std::vector<nvinfer1::Dims> GetDimVec(std::vector<TensorShapeType> shape_vec) { in GetDimVec()
71 for (auto& shape_vec : collected_shapes) { in ImplicitBatchModeCompatibleStrategy() local
127 for (auto& shape_vec : collected_shapes) { in OptimalStrategy() local
260 auto shape_vec = input_shapes_[i]; in InitProfiles() local
|
| /external/tensorflow/tensorflow/core/profiler/internal/ |
| D | tfprof_node.cc | 258 std::vector<int64> shape_vec; in ShapeProtoToVec() local
270 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 | 299 auto shape_vec = tensor_in_shape.vec<int32>(); in Compute() local
458 auto shape_vec = tensor_in_shape.vec<int32>(); in Compute() local
544 auto shape_vec = tensor_in_shape.vec<int32>(); in Compute() local
|
| D | sparse_dense_binary_op_shared.cc | 88 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 | 643 auto shape_vec = shape_tensor.flat<int32>(); in Compute() local
|
| D | pooling_ops_3d.cc | 514 auto shape_vec = tensor_in_shape.vec<int32>(); in Compute() local
|
| D | sparse_cross_op.cc | 725 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/compiler/mlir/hlo/lib/utils/ |
| D | codegen_utils.cc | 95 SmallVector<Value, 4> shape_vec; in calcMultiDimIndex() local
|
| /external/tensorflow/tensorflow/lite/tools/versioning/ |
| D | op_signature.cc | 65 const flatbuffers::Vector<int32_t>* shape_vec = in GetOpSignatureTensorSpecs() local
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/kernels/ |
| D | trt_engine_resource_ops_test.cc | 164 std::vector<TensorShape> shape_vec(n_input); in CreateTRTEngine() local
|
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_avgpooling_op.cc | 195 auto shape_vec = orig_input_tensor.vec<int32>(); in Compute() local
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | common_shape_fns.cc | 2462 auto shape_vec = shape_tensor->flat<int64>(); in SparseReduceShapeFn() local
|
| /external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/ |
| D | legalize_tf.cc | 1012 SmallVector<int64_t> shape_vec; in matchAndRewrite() local
|
| D | legalize_common.cc | 2432 SmallVector<int64_t> shape_vec(input_shape.begin(), input_shape.end()); in convertReduceOpCommon() local
|
| D | legalize_tfl.cc | 1079 SmallVector<int64_t> shape_vec; in matchAndRewrite() local
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_nodes_test.cc | 1212 std::vector<TensorShape>* shape_vec) { in GetShapeFromDataVec()
|