| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/ |
| H A D | dynamic_update_slice_util.cc | 96 const Shape& update_shape, const IndexGenerator& start_indices_generator, in EmitDynamicUpdateSliceInPlaceImpl()
171 Shape update_shape = update_array.GetShape(); in EmitDynamicUpdateSliceInPlace() local
201 Shape update_shape = update->shape(); in EmitFusedDynamicUpdateSliceInPlaceImpl() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/ |
| H A D | dynamic_update_slice.cc | 97 const RuntimeShape& update_shape) { in ClampStartIndices()
111 const auto& update_shape = GetTensorShape(update); in DynamicUpdateSlice() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| H A D | dynamic_slice_ops.cc | 50 const TensorShape update_shape = ctx->InputShape("update"); in Compile() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/ |
| H A D | hlo_module_util.cc | 153 auto update_shape = [&shape_representation_fn, in UpdateEntryComputationLayout() local
|
| H A D | hlo_alias_analysis_test.cc | 901 Shape update_shape = ShapeUtil::MakeShape(F32, {4}); in TEST_F() local
|
| H A D | hlo_sharding_util.cc | 1119 const Shape& update_shape, in GetScatterSliceSize()
|
| H A D | shape_inference.cc | 2693 const Shape& operand_shape, const Shape& update_shape, in InferDynamicUpdateSliceShape()
|
| H A D | dynamic_dimension_inference.cc | 1654 const Shape& update_shape = hlo->operand(2)->shape(); in HandleScatter() local
|
| H A D | hlo_dataflow_analysis_test.cc | 2475 Shape update_shape = ShapeUtil::MakeShape(F32, {1, 4}); in TEST_F() local
|
| H A D | algebraic_simplifier.cc | 5051 auto update_shape = dus_update->shape(); in HandleDynamicUpdateSlice() local
|
| H A D | algebraic_simplifier_test.cc | 6131 const Shape update_shape = ShapeUtil::MakeShape(F32, {0}); in TEST_F() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| H A D | spmd_partitioner.cc | 3120 const auto& update_shape = replicate_update->shape(); in HandleDynamicUpdateSlice() local
4536 auto update_shape = [this](Shape* subshape, const xla::ShapeIndex& index) { in Run() local
|
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_nn.py | 3597 def update_shape(shape, dim, new_dim_size): function
3725 def update_shape(shape, dim, new_dim_size): function
3791 def update_shape(shape, dim, new_dim_size): function
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| H A D | legalize_hlo.cc | 1214 auto update_shape = update_type.getShape(); in matchAndRewrite() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| H A D | ir_emitter_unnested.cc | 5157 Shape update_shape = in EmitDynamicUpdateSlice() local
|