Home
last modified time | relevance | path

Searched defs:tuple_shape (Results 1 – 25 of 36) 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
271 const Shape tuple_shape = in TEST_F() local
329 const Shape tuple_shape = in TEST_F() local
515 const Shape tuple_shape = in TEST_F() local
610 const Shape tuple_shape = in TEST_F() local
683 const Shape tuple_shape = in TEST_F() local
782 const Shape tuple_shape = in TEST_F() local
869 const Shape tuple_shape = tuple1->shape(); in TEST_F() local
932 const Shape tuple_shape = ShapeUtil::MakeTupleShape({scalar_shape_}); in TEST_F() local
Dmemory_space_assignment_test.cc742 Shape tuple_shape = ShapeUtil::MakeTupleShape({shape, scalar_shape}); in TEST_P() local
837 Shape tuple_shape = in TEST_P() local
1027 Shape tuple_shape = ShapeUtil::MakeTupleShape({shape, shape}); in TEST_P() local
1178 Shape tuple_shape = ShapeUtil::MakeTupleShape({shape, shape}); in TEST_P() local
2528 Shape tuple_shape = ShapeUtil::MakeTupleShape({shape3, shape4}); in TEST_P() local
2625 Shape tuple_shape = ShapeUtil::MakeTupleShape({shape, scalar_shape}); in TEST_P() local
2948 Shape tuple_shape = in TEST_P() local
3040 Shape tuple_shape = ShapeUtil::MakeTupleShape({shape, scalar_shape, shape}); in TEST_P() local
3171 Shape tuple_shape = ShapeUtil::MakeTupleShape({shape, shape}); in TEST_P() local
3215 Shape tuple_shape = ShapeUtil::MakeTupleShape({shape, shape}); in TEST_P() local
[all …]
Dhlo_graph_dumper_test.cc129 Shape tuple_shape = ShapeUtil::MakeTupleShape( in TEST_F() local
Dhlo_matchers_test.cc168 auto tuple_shape = ShapeUtil::MakeTupleShape( in TEST_F() local
Dhlo_sharding_test.cc406 auto tuple_shape = in TEST_P() local
417 auto tuple_shape = ShapeUtil::MakeTupleShape( in TEST_P() local
Dhlo_dataflow_analysis_test.cc389 const Shape tuple_shape = in TEST_P() local
492 const Shape tuple_shape = in TEST_P() local
567 const Shape tuple_shape = ShapeUtil::MakeTupleShape({scalar_shape_}); in TEST_P() local
660 const Shape tuple_shape = in TEST_P() local
776 const Shape tuple_shape = in TEST_P() local
829 const Shape tuple_shape = in TEST_P() local
949 const Shape tuple_shape = tuple1->shape(); in TEST_P() local
1075 const Shape tuple_shape = in TEST_P() local
Dshaped_buffer_test.cc156 Shape tuple_shape = in TEST() local
Dhlo_sharding.cc125 HloSharding HloSharding::Tuple(const Shape& tuple_shape, in Tuple()
138 HloSharding HloSharding::SingleTuple(const Shape& tuple_shape, in SingleTuple()
Dcopy_insertion_test.cc1576 const Shape tuple_shape = in TEST_F() local
1609 const Shape tuple_shape = in TEST_F() local
1642 const Shape tuple_shape = in TEST_F() local
1675 const Shape tuple_shape = in TEST_F() local
1710 const Shape tuple_shape = in TEST_F() local
1748 const Shape tuple_shape = in TEST_F() local
Dhlo_live_range_test.cc267 Shape tuple_shape = ShapeUtil::MakeTupleShape({shape, scalar_shape}); in TEST_F() local
Dtuple_points_to_analysis_test.cc440 Shape tuple_shape = ShapeUtil::MakeTupleShape( in TEST_F() local
679 Shape tuple_shape = in Run() local
Dconditional_simplifier.cc238 Shape tuple_shape = ShapeUtil::MakeTupleShape(new_tuple_shapes); in TryRemoveUnusedConditionalOperands() local
Dtriangular_solve_expander.cc309 Shape tuple_shape = ShapeUtil::MakeTupleShape(tuple_shapes); in InvertDiagonalBlocks() local
Dpattern_matcher_test.cc106 auto tuple_shape = ShapeUtil::MakeTupleShape({ in TEST_F() local
/external/tensorflow/tensorflow/core/tpu/kernels/xla/
Doutfeed_ops.cc72 xla::Shape tuple_shape = xla::ShapeUtil::MakeTupleShape(xla_shapes); in Compile() 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.cc372 const Shape tuple_shape = in XLA_TEST_F() local
416 const Shape tuple_shape = ShapeUtil::MakeTupleShape(element_shapes); in XLA_TEST_F() local
458 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/service/llvm_ir/
Dtuple_ops.cc85 const Shape& tuple_shape, llvm::IRBuilder<>* b) { in EmitTupleAllocasAtFunctionEntry()
/external/tensorflow/tensorflow/compiler/xla/
Dshape_util_test.cc446 Shape tuple_shape = in TEST() local
481 Shape tuple_shape = ShapeUtil::MakeTupleShape({array_shape, array_shape}); in TEST() local
737 Shape tuple_shape = ShapeUtil::MakeTupleShape({shape}); in TEST() local
/external/tensorflow/tensorflow/compiler/xla/python/tpu_driver/client/
Dtpu_client.cc174 std::vector<BorrowingLiteral> leaves, const Shape& tuple_shape, in FromLiterals()
253 Shape tuple_shape = ShapeUtil::MakeTupleShape(child_shapes); in MakeTuple() local
/external/tensorflow/tensorflow/core/tpu/
Dtpu_execute.cc140 const xla::Shape& tuple_shape, in FixTupleTableAsync()
/external/tensorflow/tensorflow/compiler/xla/tools/
Ddriver.cc470 TupleShape tuple_shape = assignment.buffers_shape[allocation_idx]; in main() local

12