Searched refs:current_shape (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | sparse_concat_op.cc | 86 const TensorShape current_shape(shapes[i].vec<int64>()); in Compute() local
88 context, current_shape.dims() == input_rank, in Compute()
91 " but got ", current_shape.dims(), " at position ", i)); in Compute()
95 context, input_shape.dim_size(j) == current_shape.dim_size(j), in Compute()
98 " for dimension ", j, " but got ", current_shape.dim_size(j), in Compute()
126 const TensorShape current_shape(shapes[i].vec<int64>()); in Compute() local
131 current_shape, std_order, &tensor)); in Compute()
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/utils/ |
| D | trt_shape_optimization_profiles.h | 100 auto current_shape = shapes[i]; in IncludesShapes() local
103 if (min[i].nbDims != current_shape.dims()) { in IncludesShapes()
107 for (int dim = 0; dim < current_shape.dims(); dim++) { in IncludesShapes()
108 if ((min[i].d[dim] > current_shape.dim_size(dim)) || in IncludesShapes()
109 (max[i].d[dim] < current_shape.dim_size(dim))) { in IncludesShapes()
|
| /external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
| D | identify_nearest_upsample.cc | 227 std::vector<int> current_shape = imagined_original_shape; in Run() local
255 current_shape.insert(current_shape.begin() + axis, scale); in Run()
256 CopyShape(current_shape, &pack_output_array); in Run()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | mhlo_to_lhlo_with_xla.h | 194 const xla::Shape& current_shape,
209 const xla::HloInstruction* instr, const xla::Shape& current_shape,
|
| D | mhlo_to_lhlo_with_xla.cc | 1330 const xla::HloInstruction* instr, const xla::Shape& current_shape, in GetOrCreateArrayView() argument
1348 const Shape static_shape = xla::ShapeUtil::MakeStaticShape(current_shape); in GetOrCreateArrayView()
1399 const HloInstruction* instr, const Shape& current_shape, in GetOrCreateViewImpl() argument
1401 if (current_shape.IsTuple()) { in GetOrCreateViewImpl()
1402 for (int i = 0; i < current_shape.tuple_shapes().size(); ++i) { in GetOrCreateViewImpl()
1405 instr, current_shape.tuple_shapes(i), current_shape_index, values)); in GetOrCreateViewImpl()
1410 if (current_shape.IsArray()) { in GetOrCreateViewImpl()
1411 TF_ASSIGN_OR_RETURN(auto v, GetOrCreateArrayView(instr, current_shape, in GetOrCreateViewImpl()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/ir/ |
| D | tfl_ops.cc | 68 SmallVector<int64_t, 4> current_shape; in VerifyOperandsHaveSameShapesOrBroadcastableShape() local
89 current_shape.assign(shape.begin(), shape.end()); in VerifyOperandsHaveSameShapesOrBroadcastableShape()
99 if (!OpTrait::util::getBroadcastedShape(current_shape, shape, in VerifyOperandsHaveSameShapesOrBroadcastableShape()
103 current_shape = result_shape; in VerifyOperandsHaveSameShapesOrBroadcastableShape()
|