Home
last modified time | relevance | path

Searched defs:tuple_shape (Results 1 – 25 of 30) sorted by relevance

12

/external/tensorflow/tensorflow/compiler/xla/service/cpu/tests/
Dcpu_bytesizeof_test.cc25 auto tuple_shape = in TEST_F() local
33 auto tuple_shape = in TEST_F() local
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_alias_analysis_test.cc222 const Shape tuple_shape = in TEST_F() local
274 const Shape tuple_shape = in TEST_F() local
335 const Shape tuple_shape = in TEST_F() local
523 const Shape tuple_shape = in TEST_F() local
618 const Shape tuple_shape = in TEST_F() local
691 const Shape tuple_shape = in TEST_F() local
790 const Shape tuple_shape = in TEST_F() local
877 const Shape tuple_shape = tuple1->shape(); in TEST_F() local
940 const Shape tuple_shape = ShapeUtil::MakeTupleShape({scalar_shape_}); in TEST_F() local
Dhlo_sharding_test.cc285 auto tuple_shape = in TEST_F() local
295 auto tuple_shape = ShapeUtil::MakeTupleShape( in TEST_F() local
Dhlo_graph_dumper_test.cc129 Shape tuple_shape = ShapeUtil::MakeTupleShape( in TEST_F() local
Dhlo_matchers_test.cc153 auto tuple_shape = ShapeUtil::MakeTupleShape( in TEST() local
Dhlo_sharding.cc59 HloSharding HloSharding::Tuple(const Shape& tuple_shape, in Tuple()
72 HloSharding HloSharding::SingleTuple(const Shape& tuple_shape, in SingleTuple()
Dcopy_insertion_test.cc1366 const Shape tuple_shape = in TEST_F() local
1401 const Shape tuple_shape = in TEST_F() local
1436 const Shape tuple_shape = in TEST_F() local
1469 const Shape tuple_shape = in TEST_F() local
1505 const Shape tuple_shape = in TEST_F() local
1544 const Shape tuple_shape = in TEST_F() local
Dhlo_dataflow_analysis_test.cc482 const Shape tuple_shape = in TEST_P() local
585 const Shape tuple_shape = in TEST_P() local
659 const Shape tuple_shape = in TEST_P() local
776 const Shape tuple_shape = in TEST_P() local
896 const Shape tuple_shape = tuple1->shape(); in TEST_P() local
1022 const Shape tuple_shape = in TEST_P() local
Dtriangular_solve_expander.cc188 Shape tuple_shape = ShapeUtil::MakeTupleShape(tuple_shapes); in InvertDiagonalBlocks() local
Dheap_simulator_test.cc44 const Shape tuple_shape = in TEST_F() local
696 const Shape tuple_shape = in TEST_F() local
Dpattern_matcher_test.cc130 auto tuple_shape = ShapeUtil::MakeTupleShape({ in TEST() local
Dtuple_points_to_analysis_test.cc412 Shape tuple_shape = ShapeUtil::MakeTupleShape( in TEST_F() local
628 Shape tuple_shape = in Run() local
Ddynamic_dimension_inference_test.cc465 auto tuple_shape = ShapeUtil::MakeTupleShape({input_shape, input_shape}); in TEST_F() local
Dbuffer_assignment_test.cc1387 auto tuple_shape = ShapeUtil::MakeTupleShape({elem_shape}); in TEST_F() local
1428 auto tuple_shape = ShapeUtil::MakeTupleShape({elem_shape}); in TEST_F() local
1513 auto tuple_shape = in TEST_F() local
Dshape_inference_test.cc888 Shape tuple_shape = ShapeUtil::MakeTupleShape({f32_, s32_}); in TEST_F() local
900 Shape tuple_shape = ShapeUtil::MakeTupleShape({f32_, s32_}); in TEST_F() local
1479 Shape tuple_shape = ShapeUtil::MakeTupleShape({input_shape, input_shape}); in TEST_F() local
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dcudnn_batchnorm_thunk.cc158 for (const auto& tuple_shape : hlo->shape().tuple_shapes()) { in CudnnBatchNormForwardTrainingThunk() local
242 for (const auto& tuple_shape : hlo->shape().tuple_shapes()) { in CudnnBatchNormBackwardThunk() local
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dcpu_copy_insertion_test.cc110 const Shape tuple_shape = in TEST_F() local
/external/tensorflow/tensorflow/compiler/xla/tests/
Dconditional_test.cc80 const Shape& tuple_shape) { in CreateTupleCeilComputation()
102 const Shape& tuple_shape) { in CreateTupleFloorComputation()
124 const Shape& tuple_shape) { in CreateTupleAddComputation()
144 const Shape& tuple_shape) { in CreateTupleSubComputation()
635 Shape tuple_shape = ShapeUtil::MakeTupleShape({r0bool, r0f32_, r0f32_}); in XLA_TEST_F() local
665 Shape tuple_shape = ShapeUtil::MakeTupleShape({r0bool, r0f32_, r0f32_}); in XLA_TEST_F() local
705 Shape tuple_shape = ShapeUtil::MakeTupleShape({r0f32_, r0f32_}); in XLA_TEST_F() local
Dlocal_client_execute_test.cc369 const Shape tuple_shape = in XLA_TEST_F() local
413 const Shape tuple_shape = ShapeUtil::MakeTupleShape(element_shapes); in XLA_TEST_F() local
455 const Shape tuple_shape = ShapeUtil::MakeTupleShape(inner_tuple_shapes); in XLA_TEST_F() local
Dparams_test.cc437 Shape tuple_shape = ShapeUtil::MakeTupleShape({r1f32_3, r1f32_3}); in XLA_TEST_F() local
/external/tensorflow/tensorflow/compiler/xla/client/lib/
Dloops.cc37 Shape tuple_shape = ShapeUtil::MakeTupleShape(var_shapes); in WhileLoopHelper() local
/external/tensorflow/tensorflow/compiler/xla/python/
Dxrt.cc256 const Shape& tuple_shape = allocation->shape(); in DestructureXrtAllocationTuple() local
Dlocal_computation_builder.cc166 const Shape tuple_shape = shape(); in DestructureTuple() local
/external/tensorflow/tensorflow/compiler/xla/
Dshape_util_test.cc422 Shape tuple_shape = in TEST() local
457 Shape tuple_shape = ShapeUtil::MakeTupleShape({array_shape, array_shape}); in TEST() local
/external/tensorflow/tensorflow/compiler/tf2xla/
Dxla_compiler.cc663 std::vector<xla::Shape> tuple_shape( in XLAShapeForArgument() local

12