| /external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/ |
| D | dynamic_update_slice_util.cc | 42 const Shape& update_shape, const IndexGenerator& start_indices_generator, in EmitDynamicUpdateSliceInPlaceImpl() argument
56 start_multi_index[i]->getType(), update_shape.dimensions(i)); in EmitDynamicUpdateSliceInPlaceImpl()
99 return gpu::ParallelLoopEmitter(loop_body_emitter, update_shape, in EmitDynamicUpdateSliceInPlaceImpl()
103 return LoopEmitter(loop_body_emitter, update_shape, b).EmitLoop(name); in EmitDynamicUpdateSliceInPlaceImpl()
117 Shape update_shape = update_array.GetShape(); in EmitDynamicUpdateSliceInPlace() local
129 update_shape, start_indices_generator, is_signed, update_array_generator, in EmitDynamicUpdateSliceInPlace()
150 Shape update_shape = update->shape(); in EmitFusedDynamicUpdateSliceInPlaceImpl() local
166 LayoutUtil::CopyLayoutBetweenShapes(fusion->shape(), &update_shape)); in EmitFusedDynamicUpdateSliceInPlaceImpl()
181 update_shape, start_indices_generator, is_signed, update_array_generator, in EmitFusedDynamicUpdateSliceInPlaceImpl()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | dynamic_slice_ops.cc | 41 const TensorShape update_shape = ctx->InputShape("update"); in Compile() local
51 ctx, input_shape.dims() == update_shape.dims(), in Compile()
55 update_shape.DebugString())); in Compile()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | tuple_points_to_analysis_test.cc | 626 Shape update_shape = ShapeUtil::MakeShape(F32, {3}); in Run() local
629 ShapeUtil::MakeTupleShape({input_shape, update_shape, starts_shape}); in Run()
637 HloInstruction::CreateGetTupleElement(update_shape, tuple_param0, 1)); in Run()
642 update_shape, HloOpcode::kAdd, tuple_element1, ones)); in Run()
650 HloInstruction::CreateSlice(update_shape, input, {0}, {3}, {1})); in Run()
653 update_shape, HloOpcode::kAdd, update, slice)); in Run()
1006 Shape update_shape = ShapeUtil::MakeShape(F32, {4}); in TEST_F() local
1011 HloInstruction::CreateParameter(1, update_shape, "update")); in TEST_F()
|
| D | buffer_liveness_test.cc | 619 Shape update_shape = ShapeUtil::MakeShape(F32, {3}); in BuildModule() local
635 HloInstruction::CreateSlice(update_shape, gte1, {0}, {3}, {1})); in BuildModule()
637 update_shape, HloOpcode::kAdd, update, slice)); in BuildModule()
777 Shape update_shape = ShapeUtil::MakeShape(F32, {3}); in Run() local
|
| D | shape_inference.cc | 2368 const Shape& operand_shape, const Shape& update_shape, in InferDynamicUpdateSliceShape() argument
2373 ExpectArray(update_shape, "update of dynamic update slice")); in InferDynamicUpdateSliceShape()
2393 ShapeUtil::HumanString(update_shape)); in InferDynamicUpdateSliceShape()
2417 ShapeUtil::HumanString(update_shape)); in InferDynamicUpdateSliceShape()
2450 if (update_shape.rank() != operand_shape.rank()) { in InferDynamicUpdateSliceShape()
2454 update_shape.rank(), operand_shape.rank()); in InferDynamicUpdateSliceShape()
2458 update_shape)) { in InferDynamicUpdateSliceShape()
2463 PrimitiveType_Name(update_shape.element_type())); in InferDynamicUpdateSliceShape()
2468 const int64 update_dim_size = update_shape.dimensions(dim); in InferDynamicUpdateSliceShape()
|
| D | shape_inference.h | 192 const Shape& operand_shape, const Shape& update_shape,
|
| D | hlo_dataflow_analysis_test.cc | 2380 Shape update_shape = ShapeUtil::MakeShape(F32, {4}); in TEST_F() local
2385 HloInstruction::CreateParameter(1, update_shape, "update")); in TEST_F()
|
| D | algebraic_simplifier_test.cc | 4590 const Shape update_shape = ShapeUtil::MakeShape(F32, {0}); in TEST_F() local
4592 HloInstruction::CreateParameter(1, update_shape, "update")); in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.cc | 814 TF_ASSIGN_OR_RETURN(const Shape& update_shape, GetShape(update)); in DynamicUpdateSlice()
819 operand_shape, update_shape, {start_indices_shape})); in DynamicUpdateSlice()
833 TF_ASSIGN_OR_RETURN(const Shape& update_shape, GetShape(update)); in DynamicUpdateSlice()
842 operand_shape, update_shape, start_indices_shapes)); in DynamicUpdateSlice()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | ir_emitter_unnested.cc | 658 Shape update_shape = root->operand(1)->shape(); in HandleFusion() local
665 update_shape, ir_emitter_context_->device_description()); in HandleFusion()
|