Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dsparse_concat_op.cc72 const TensorShape current_shape(shapes[i].vec<int64_t>()); in operator ()() local
164 const TensorShape current_shape(shapes[i].vec<int64_t>()); in Compute() local
H A Dsparse_concat_op_gpu.cu.cc150 const TensorShape current_shape(shapes[i].vec<int64_t>()); in operator ()() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2tensorrt/utils/
H A Dtrt_shape_optimization_profiles.h154 auto current_shape = shapes[i]; in IncludesShapes() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/toco/graph_transformations/
H A Didentify_nearest_upsample.cc228 std::vector<int> current_shape = imagined_original_shape; in Run() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
H A Dmhlo_to_lhlo_with_xla.cc1427 const xla::HloInstruction* instr, const xla::Shape& current_shape, in GetOrCreateArrayView()
1494 const HloInstruction* instr, const Shape& current_shape, in GetOrCreateViewImpl()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/ir/
H A Dtfl_ops.cc161 SmallVector<int64_t, 4> current_shape; in VerifyOperandsHaveSameShapesOrBroadcastableShape() local