| /external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/ |
| D | dynamic_update_slice_util.cc | 96 const Shape& update_shape, const IndexGenerator& start_indices_generator, in EmitDynamicUpdateSliceInPlaceImpl() argument
110 start_multi_index[i]->getType(), update_shape.dimensions(i)); in EmitDynamicUpdateSliceInPlaceImpl()
153 return gpu::ParallelLoopEmitter(loop_body_emitter, update_shape, in EmitDynamicUpdateSliceInPlaceImpl()
157 return LoopEmitter(loop_body_emitter, update_shape, b).EmitLoop(name); in EmitDynamicUpdateSliceInPlaceImpl()
171 Shape update_shape = update_array.GetShape(); in EmitDynamicUpdateSliceInPlace() local
183 update_shape, start_indices_generator, is_signed, update_array_generator, in EmitDynamicUpdateSliceInPlace()
201 Shape update_shape = update->shape(); in EmitFusedDynamicUpdateSliceInPlaceImpl() local
217 dynamic_update_slice->shape(), &update_shape)); in EmitFusedDynamicUpdateSliceInPlaceImpl()
232 update_shape, start_indices_generator, is_signed, update_array_generator, in EmitFusedDynamicUpdateSliceInPlaceImpl()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | dynamic_slice_ops.cc | 50 const TensorShape update_shape = ctx->InputShape("update"); in Compile() local
61 ctx, rank == update_shape.dims(), in Compile()
65 update_shape.DebugString())); in Compile()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | tuple_points_to_analysis_test.cc | 677 Shape update_shape = ShapeUtil::MakeShape(F32, {3}); in Run() local
680 ShapeUtil::MakeTupleShape({input_shape, update_shape, starts_shape}); in Run()
688 HloInstruction::CreateGetTupleElement(update_shape, tuple_param0, 1)); in Run()
693 update_shape, HloOpcode::kAdd, tuple_element1, ones)); in Run()
701 HloInstruction::CreateSlice(update_shape, input, {0}, {3}, {1})); in Run()
704 update_shape, HloOpcode::kAdd, update, slice)); in Run()
|
| D | shape_inference.cc | 2576 const Shape& operand_shape, const Shape& update_shape, in InferDynamicUpdateSliceShape() argument
2581 ExpectArray(update_shape, "update of dynamic update slice")); in InferDynamicUpdateSliceShape()
2601 ShapeUtil::HumanString(update_shape)); in InferDynamicUpdateSliceShape()
2625 ShapeUtil::HumanString(update_shape)); in InferDynamicUpdateSliceShape()
2658 if (update_shape.rank() != operand_shape.rank()) { in InferDynamicUpdateSliceShape()
2662 update_shape.rank(), operand_shape.rank()); in InferDynamicUpdateSliceShape()
2666 update_shape)) { in InferDynamicUpdateSliceShape()
2671 PrimitiveType_Name(update_shape.element_type())); in InferDynamicUpdateSliceShape()
2676 const int64 update_dim_size = update_shape.dimensions(dim); in InferDynamicUpdateSliceShape()
2696 for (int64 i = 0; i < update_shape.rank(); ++i) { in InferDynamicUpdateSliceShape()
[all …]
|
| D | shape_inference.h | 211 const Shape& operand_shape, const Shape& update_shape,
|
| D | hlo_alias_analysis_test.cc | 1067 Shape update_shape = ShapeUtil::MakeShape(F32, {4}); in TEST_F() local
1073 HloInstruction::CreateParameter(1, update_shape, "param1")); in TEST_F()
|
| D | hlo_dataflow_analysis_test.cc | 2476 Shape update_shape = ShapeUtil::MakeShape(F32, {1, 4}); in TEST_F() local
2481 HloInstruction::CreateParameter(1, update_shape, "update")); in TEST_F()
|
| D | algebraic_simplifier.cc | 4435 auto update_shape = dus_update->shape(); in HandleDynamicUpdateSlice() local
4470 auto update_width = ShapeUtil::GetDimension(update_shape, dim); in HandleDynamicUpdateSlice()
|
| D | algebraic_simplifier_test.cc | 5471 const Shape update_shape = ShapeUtil::MakeShape(F32, {0}); in TEST_F() local
5473 HloInstruction::CreateParameter(1, update_shape, "update")); in TEST_F()
|
| /external/tensorflow/tensorflow/python/keras/ |
| D | backend.py | 5055 update_shape = (
5058 if update_shape:
5070 if update_shape and output_rank >= 3:
|
| /external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| D | spmd_partitioner.cc | 2374 const auto& update_shape = replicate_update->shape(); in HandleDynamicUpdateSlice() local
2388 ((partitioned_slice_offsets[i] + update_shape.dimensions(dim) - 1) / in HandleDynamicUpdateSlice()
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.cc | 1031 TF_ASSIGN_OR_RETURN(const Shape* update_shape, GetShapePtr(update)); in DynamicUpdateSlice()
1040 *operand_shape, *update_shape, start_indices_shapes)); in DynamicUpdateSlice()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | ir_emitter_unnested.cc | 2036 Shape update_shape = fusion->fused_expression_root()->operand(1)->shape(); in HandleFusion() local
2043 update_shape, ir_emitter_context_->gpu_device_info()); in HandleFusion()
|