| /external/tensorflow/tensorflow/compiler/xla/ |
| D | shape_util_test.cc | 356 ShapeUtil::IsNestedTuple(ShapeUtil::MakeTupleShapeWithPtrs({&s32}))); in TEST()
358 ShapeUtil::IsNestedTuple(ShapeUtil::MakeTupleShapeWithPtrs({&nil}))); in TEST()
360 ShapeUtil::MakeTupleShapeWithPtrs({&s32, &s32}))); in TEST()
362 ShapeUtil::MakeTupleShapeWithPtrs({&s32, &nil}))); in TEST()
364 ShapeUtil::MakeTupleShapeWithPtrs({&nil, &s32}))); in TEST()
366 ShapeUtil::MakeTupleShapeWithPtrs({&nil, &nil}))); in TEST()
|
| D | literal_util.cc | 546 Literal literal(ShapeUtil::MakeTupleShapeWithPtrs(element_shapes)); in MakeTuple()
560 Literal literal(ShapeUtil::MakeTupleShapeWithPtrs(element_shapes)); in MakeTupleFromSlices()
574 Literal literal(ShapeUtil::MakeTupleShapeWithPtrs(element_shapes)); in MakeTupleOwned()
|
| D | shape_util.h | 274 static Shape MakeTupleShapeWithPtrs(absl::Span<const Shape* const> shapes);
|
| D | shape_util.cc | 406 /* static */ Shape ShapeUtil::MakeTupleShapeWithPtrs( in MakeTupleShapeWithPtrs() function in xla::ShapeUtil
|
| D | literal.cc | 1755 Literal literal(ShapeUtil::MakeTupleShapeWithPtrs(element_shapes), in MoveIntoTuple()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | all_reduce_combiner.cc | 84 ShapeUtil::MakeTupleShapeWithPtrs(operand_shapes), operands, reduction, in CombineAllReduces()
|
| D | rng_bit_generator_expander.cc | 116 ShapeUtil::MakeTupleShapeWithPtrs({&state_shape, &data_shape}), in ExpandInstruction()
|
| D | reduce_scatter_combiner.cc | 88 ShapeUtil::MakeTupleShapeWithPtrs(output_shapes), operands, reduction, in CombineReduceScatters()
|
| D | all_gather_combiner.cc | 80 ShapeUtil::MakeTupleShapeWithPtrs(output_shapes), operands, in CombineAllGathers()
|
| D | all_to_all_decomposer.cc | 113 Shape all_to_all_shape = ShapeUtil::MakeTupleShapeWithPtrs( in ExpandInstruction()
|
| D | sort_simplifier.cc | 80 : ShapeUtil::MakeTupleShapeWithPtrs(new_shapes); in RemoveUnusedOperandFromSort()
|
| D | async_collective_creator.cc | 91 ? ShapeUtil::MakeTupleShapeWithPtrs(operand_shapes) in Run()
|
| D | conditional_simplifier.cc | 104 Shape tuple_shape = ShapeUtil::MakeTupleShapeWithPtrs(new_tuple_shapes); in TryRemoveUnusedConditionalOperands()
254 const Shape new_shape = ShapeUtil::MakeTupleShapeWithPtrs(new_tuple_shapes); in RemoveUnusedTupleElements()
|
| D | shape_inference.cc | 1105 return ShapeUtil::MakeTupleShapeWithPtrs(operand_shapes); in InferVariadicOpShape()
1309 return ShapeUtil::MakeTupleShapeWithPtrs({&operand_shape, in InferBatchNormTrainingShape()
1610 return ShapeUtil::MakeTupleShapeWithPtrs( in InferBatchNormGradShape()
2069 input_shape = ShapeUtil::MakeTupleShapeWithPtrs(operand_shapes); in InferAllGatherStartShape()
2071 return ShapeUtil::MakeTupleShapeWithPtrs({&input_shape, &ag_shape}); in InferAllGatherStartShape()
2088 return ShapeUtil::MakeTupleShapeWithPtrs(operand_shapes); in InferAllReduceShape()
2201 return ShapeUtil::MakeTupleShapeWithPtrs( in InferCollectivePermuteStartShape()
2205 return ShapeUtil::MakeTupleShapeWithPtrs( in InferCollectivePermuteStartShape()
3783 : ShapeUtil::MakeTupleShapeWithPtrs(operand_shapes); in InferScatterShape()
|
| D | while_loop_simplifier.cc | 84 ShapeUtil::MakeTupleShapeWithPtrs(new_while_tuple_elem_shapes); in RemoveDeadTupleIndices()
670 ShapeUtil::MakeTupleShapeWithPtrs(new_while_shape_elems); in TryRemoveConstantParams()
1026 ShapeUtil::MakeTupleShapeWithPtrs(flattened_shape_elems); in TryFlattenNestedTuples()
|
| D | hlo_creation_utils_test.cc | 348 ShapeUtil::MakeTupleShapeWithPtrs({&input_shape1, &input_shape0}); in TEST_F()
|
| D | hlo_instruction.cc | 1922 Shape tuple_shape = ShapeUtil::MakeTupleShapeWithPtrs(element_shapes); in CreateTuple()
|
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | approx_topk.cc | 230 ShapeUtil::MakeTupleShapeWithPtrs(approx_output_shapes); in ApproxTopK()
|
| /external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| D | gather_scatter_handler.cc | 747 return ShapeUtil::MakeTupleShapeWithPtrs(element_shapes); in MaybeMakeTupleShape()
|
| D | dot_handler.cc | 1479 ShapeUtil::MakeTupleShapeWithPtrs( in PartitionBaseCase()
1756 ShapeUtil::MakeTupleShapeWithPtrs( in PartitionBaseCase()
|
| D | spmd_partitioner.cc | 2851 auto operand_shape = ShapeUtil::MakeTupleShapeWithPtrs(operand_shapes); in HandleSingleDevice()
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.cc | 2353 ShapeUtil::MakeTupleShapeWithPtrs({&state_shape, &output_shape}), in RngBitGenerator()
|