Home
last modified time | relevance | path

Searched refs:shape_vec (Results 1 – 21 of 21) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2tensorrt/utils/
Dtrt_shape_optimization_profiles.cc30 std::vector<nvinfer1::Dims> GetDimVec(std::vector<TensorShapeType> shape_vec) { in GetDimVec() argument
31 std::vector<nvinfer1::Dims> dimvec(shape_vec.size()); in GetDimVec()
32 absl::c_transform(shape_vec, dimvec.begin(), [](TensorShapeType shape) { in GetDimVec()
62 for (auto& shape_vec : input_shapes_) { in ImplicitBatchModeCompatibleStrategy() local
63 if (!shape_vec.empty()) { in ImplicitBatchModeCompatibleStrategy()
64 std::vector<nvinfer1::Dims> dimvec = GetDimVec(shape_vec); in ImplicitBatchModeCompatibleStrategy()
74 for (auto& shape_vec : input_shapes_) { in OptimalStrategy() local
75 if (!shape_vec.empty()) { in OptimalStrategy()
76 std::vector<nvinfer1::Dims> min = GetDimVec(shape_vec); in OptimalStrategy()
Dtrt_shape_optimization_profiles_test.cc151 std::vector<TensorShape> shape_vec = DimVecToShapeVec(dim_vec); in TEST_F() local
152 EXPECT_EQ(0, profile.GetProfileNumber(shape_vec)); in TEST_F()
170 std::vector<TensorShape> shape_vec = DimVecToShapeVec(dim_vec); in TEST_F() local
171 profile.AddShape(shape_vec); in TEST_F()
192 std::vector<TensorShape> shape_vec = DimVecToShapeVec(dimvec); in TEST_F() local
193 int idx = profile.GetProfileNumber(shape_vec); in TEST_F()
/external/tensorflow/tensorflow/core/profiler/internal/
Dtfprof_node.cc261 std::vector<int64> shape_vec; in ShapeProtoToVec() local
264 shape_vec.push_back(1); in ShapeProtoToVec()
267 shape_vec.push_back(d.size()); in ShapeProtoToVec()
270 return shape_vec; in ShapeProtoToVec()
273 TensorShapeProto VecToShapeProto(const std::vector<int64>& shape_vec) { in VecToShapeProto() argument
275 if (shape_vec.empty()) { in VecToShapeProto()
279 for (const int64 s : shape_vec) { in VecToShapeProto()
Dtfprof_show.cc227 std::vector<string> shape_vec; in FormatNode() local
230 shape_vec.push_back(absl::StrFormat("%d:unknown", s.first)); in FormatNode()
232 shape_vec.push_back( in FormatNode()
236 info.push_back(absl::StrJoin(shape_vec, "|")); in FormatNode()
Dtfprof_node.h40 TensorShapeProto VecToShapeProto(const std::vector<int64>& shape_vec);
708 std::vector<int64>& shape_vec = input_shapes[inp.first]; in input_shapes() local
722 shape_vec.assign(output_shape->second.begin(), in input_shapes()
/external/tensorflow/tensorflow/core/kernels/image/
Drandom_crop_op.cc47 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()
Dimage_ops.cc77 auto shape_vec = shape_t.vec<int32>(); in DoImageProjectiveTransformOp() local
78 out_height = shape_vec(0); in DoImageProjectiveTransformOp()
79 out_width = shape_vec(1); in DoImageProjectiveTransformOp()
/external/tensorflow/tensorflow/core/kernels/
Davgpooling_op.cc294 auto shape_vec = tensor_in_shape.vec<int32>(); in Compute() local
296 output_shape.AddDim(shape_vec(i)); in Compute()
452 auto shape_vec = tensor_in_shape.vec<int32>(); in Compute() local
454 output_shape.AddDim(shape_vec(i)); in Compute()
538 auto shape_vec = tensor_in_shape.vec<int32>(); in Compute() local
540 output_shape.AddDim(shape_vec(i)); in Compute()
Dsparse_dense_binary_op_shared.cc83 const auto shape_vec = shape_t->vec<int64>(); in Compute() local
84 const auto lhs_dims = BCast::FromShape(TensorShape(shape_vec)); in Compute()
Dsparse_reduce_op.cc174 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()
Dremote_fused_graph_execute_utils.cc857 std::vector<TensorShape> shape_vec; in BuildClusterSubgraphDef() local
858 GetOutputTensorShapeType(*node_def, &dt_vec, &shape_vec).IgnoreError(); in BuildClusterSubgraphDef()
862 shape_vec.empty() ? TensorShape({}) : shape_vec.at(subgraph_input_port); in BuildClusterSubgraphDef()
1418 std::vector<TensorShape> shape_vec; in BuildNodeMapFromOpsDefinitions() local
1420 GetOutputTensorShapeType(node_def, &dt_vec, &shape_vec); in BuildNodeMapFromOpsDefinitions()
1422 shape_vec.clear(); in BuildNodeMapFromOpsDefinitions()
Dparameterized_truncated_normal_op.cc640 auto shape_vec = shape_tensor.flat<int32>(); in Compute() local
643 shape_vec.data(), shape_vec.size(), &tensor_shape)); in Compute()
Dsparse_cross_op.cc684 auto shape_vec = (*shape_out)->vec<int64>(); in CreateOutputTensors() local
685 shape_vec(0) = batch_size; in CreateOutputTensors()
686 shape_vec(1) = max_cross_count; in CreateOutputTensors()
Dpooling_ops_3d.cc501 auto shape_vec = tensor_in_shape.vec<int32>(); in Compute() local
503 output_shape.AddDim(shape_vec(i)); in Compute()
/external/tensorflow/tensorflow/core/kernels/mkl/
Dmkl_avgpooling_op.cc194 auto shape_vec = orig_input_tensor.vec<int32>(); in Compute() local
197 orig_input_shape.AddDim(shape_vec(i)); in Compute()
/external/tensorflow/tensorflow/core/framework/
Dcommon_shape_fns.cc2422 auto shape_vec = shape_tensor->flat<int64>(); in SparseReduceShapeFn() local
2425 int64 ndims = shape_vec.size(); in SparseReduceShapeFn()
2436 dims.push_back(c->MakeDim(shape_vec(d))); in SparseReduceShapeFn()
2444 dims.push_back(c->MakeDim(shape_vec(d))); in SparseReduceShapeFn()
/external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/
Dlegalize_common.cc2330 SmallVector<int64_t, 4> shape_vec(input_shape.begin(), input_shape.end()); in convertReduceOpCommon() local
2341 shape_vec[axis_val] = 1; in convertReduceOpCommon()
2343 llvm::makeArrayRef<int64_t>(shape_vec), reduce_element_type); in convertReduceOpCommon()
2353 llvm::makeArrayRef<int64_t>(shape_vec), output_type.getElementType()); in convertReduceOpCommon()
Dlegalize_tf.cc1007 llvm::SmallVector<int64_t, 4> shape_vec; in matchAndRewrite() local
1009 shape_vec.push_back( in matchAndRewrite()
1011 output_shape = rewriter.getI64ArrayAttr(shape_vec); in matchAndRewrite()
Dlegalize_tfl.cc1247 llvm::SmallVector<int64_t, 4> shape_vec; in matchAndRewrite() local
1249 shape_vec.push_back( in matchAndRewrite()
1251 output_shape = rewriter.getI64ArrayAttr(shape_vec); in matchAndRewrite()
/external/tensorflow/tensorflow/python/framework/
Dops.py460 shape_vec, unknown_shape = pywrap_tf_session.TF_GraphGetTensorShapeHelper(
465 shape_vec = [None if d == -1 else d for d in shape_vec]
466 return tensor_shape.TensorShape(shape_vec)
/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_nodes_test.cc1323 std::vector<TensorShape>* shape_vec) { in GetShapeFromDataVec() argument
1324 shape_vec->reserve(input_data.size()); in GetShapeFromDataVec()
1326 std::back_inserter(*shape_vec), in GetShapeFromDataVec()