Home
last modified time | relevance | path

Searched defs:tuple_shape (Results 1 – 25 of 35) 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
Dmemory_space_assignment_test.cc552 Shape tuple_shape = ShapeUtil::MakeTupleShape({shape, scalar_shape}); in TEST_P() local
647 Shape tuple_shape = in TEST_P() local
834 Shape tuple_shape = ShapeUtil::MakeTupleShape({shape, shape}); in TEST_P() local
985 Shape tuple_shape = ShapeUtil::MakeTupleShape({shape, shape}); in TEST_P() local
1406 Shape tuple_shape = ShapeUtil::MakeTupleShape({shape3, shape4}); in TEST_P() local
1503 Shape tuple_shape = ShapeUtil::MakeTupleShape({shape, scalar_shape}); in TEST_P() local
1825 Shape tuple_shape = in TEST_P() local
1917 Shape tuple_shape = ShapeUtil::MakeTupleShape({shape, scalar_shape, shape}); in TEST_P() local
2054 Shape tuple_shape = ShapeUtil::MakeTupleShape({shape, shape}); in TEST_P() local
2098 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_sharding.cc59 HloSharding HloSharding::Tuple(const Shape& tuple_shape, in Tuple()
72 HloSharding HloSharding::SingleTuple(const Shape& tuple_shape, in SingleTuple()
Dhlo_matchers_test.cc168 auto tuple_shape = ShapeUtil::MakeTupleShape( in TEST_F() local
Dcopy_insertion_test.cc1377 const Shape tuple_shape = in TEST_F() local
1412 const Shape tuple_shape = in TEST_F() local
1447 const Shape tuple_shape = in TEST_F() local
1480 const Shape tuple_shape = in TEST_F() local
1516 const Shape tuple_shape = in TEST_F() local
1555 const Shape tuple_shape = in TEST_F() local
Dshaped_buffer_test.cc156 Shape tuple_shape = in TEST() 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
Dconditional_simplifier.cc229 Shape tuple_shape = ShapeUtil::MakeTupleShape(new_tuple_shapes); in TryRemoveUnusedConditionalOperands() local
Dhlo_live_range_test.cc268 Shape tuple_shape = ShapeUtil::MakeTupleShape({shape, scalar_shape}); in TEST_F() local
Dtuple_points_to_analysis_test.cc438 Shape tuple_shape = ShapeUtil::MakeTupleShape( in TEST_F() local
654 Shape tuple_shape = in Run() local
Dtriangular_solve_expander.cc188 Shape tuple_shape = ShapeUtil::MakeTupleShape(tuple_shapes); in InvertDiagonalBlocks() local
Dpattern_matcher_test.cc106 auto tuple_shape = ShapeUtil::MakeTupleShape({ in TEST_F() local
Dheap_simulator_test.cc44 const Shape tuple_shape = in TEST_F() local
884 const Shape tuple_shape = in TEST_F() local
/external/tensorflow/tensorflow/compiler/xla/python/
Dshared_device_buffer_test.cc47 Shape tuple_shape = ShapeUtil::MakeTupleShape({a_shape, b_shape}); in TEST() 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
/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.cc84 const Shape& tuple_shape, llvm::IRBuilder<>* b) { in EmitTupleAllocasAtFunctionEntry()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dcustom_call_test.cc160 auto tuple_shape = ShapeUtil::MakeTupleShape({ in TEST_F() local
/external/tensorflow/tensorflow/compiler/xla/
Dshape_util_test.cc444 Shape tuple_shape = in TEST() local
479 Shape tuple_shape = ShapeUtil::MakeTupleShape({array_shape, array_shape}); in TEST() local
738 Shape tuple_shape = ShapeUtil::MakeTupleShape({shape}); in TEST() local
/external/tensorflow/tensorflow/compiler/xla/python/tpu_driver/client/
Dtpu_client.cc175 std::vector<BorrowingLiteral> leaves, const Shape& tuple_shape, in FromLiterals()
253 Shape tuple_shape = ShapeUtil::MakeTupleShape(child_shapes); in MakeTuple() local
/external/tensorflow/tensorflow/compiler/xla/tools/
Ddriver.cc447 TupleShape tuple_shape = assignment.buffers_shape[param_idx]; in main() local

12