Home
last modified time | relevance | path

Searched defs:current_shape (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dsparse_concat_op.cc86 const TensorShape current_shape(shapes[i].vec<int64>()); in Compute() local
126 const TensorShape current_shape(shapes[i].vec<int64>()); in Compute() local
/external/tensorflow/tensorflow/compiler/tf2tensorrt/utils/
Dtrt_shape_optimization_profiles.h100 auto current_shape = shapes[i]; in IncludesShapes() local
/external/tensorflow/tensorflow/lite/toco/graph_transformations/
Didentify_nearest_upsample.cc227 std::vector<int> current_shape = imagined_original_shape; in Run() local
/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
Dmhlo_to_lhlo_with_xla.cc1330 const xla::HloInstruction* instr, const xla::Shape& current_shape, in GetOrCreateArrayView()
1399 const HloInstruction* instr, const Shape& current_shape, in GetOrCreateViewImpl()
/external/tensorflow/tensorflow/compiler/mlir/lite/ir/
Dtfl_ops.cc68 SmallVector<int64_t, 4> current_shape; in VerifyOperandsHaveSameShapesOrBroadcastableShape() local