Searched defs:update_shape (Results 1 – 8 of 8) sorted by relevance
| /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()
117 Shape update_shape = update_array.GetShape(); in EmitDynamicUpdateSliceInPlace() local
150 Shape update_shape = update->shape(); in EmitFusedDynamicUpdateSliceInPlaceImpl() local
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | dynamic_slice_ops.cc | 41 const TensorShape update_shape = ctx->InputShape("update"); in Compile() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | buffer_liveness_test.cc | 619 Shape update_shape = ShapeUtil::MakeShape(F32, {3}); in BuildModule() local
777 Shape update_shape = ShapeUtil::MakeShape(F32, {3}); in Run() local
|
| D | tuple_points_to_analysis_test.cc | 626 Shape update_shape = ShapeUtil::MakeShape(F32, {3}); in Run() local
1006 Shape update_shape = ShapeUtil::MakeShape(F32, {4}); in TEST_F() local
|
| D | shape_inference.cc | 2368 const Shape& operand_shape, const Shape& update_shape, in InferDynamicUpdateSliceShape()
|
| D | hlo_dataflow_analysis_test.cc | 2380 Shape update_shape = ShapeUtil::MakeShape(F32, {4}); in TEST_F() local
|
| D | algebraic_simplifier_test.cc | 4590 const Shape update_shape = ShapeUtil::MakeShape(F32, {0}); in TEST_F() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | ir_emitter_unnested.cc | 658 Shape update_shape = root->operand(1)->shape(); in HandleFusion() local
|