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.cc255 const Shape tuple_shape = in TEST_F() local
304 const Shape tuple_shape = in TEST_F() local
362 const Shape tuple_shape = in TEST_F() local
547 const Shape tuple_shape = in TEST_F() local
640 const Shape tuple_shape = in TEST_F() local
713 const Shape tuple_shape = in TEST_F() local
812 const Shape tuple_shape = in TEST_F() local
Dtuple_util.cc64 const Shape& tuple_shape = tuple->shape(); in ReplaceTupleWith() local
Dhlo_graph_dumper_test.cc130 Shape tuple_shape = ShapeUtil::MakeTupleShape( in TEST_F() local
Dmemory_space_assignment_test.cc777 Shape tuple_shape = ShapeUtil::MakeTupleShape({shape, scalar_shape}); in TEST_P() local
872 Shape tuple_shape = in TEST_P() local
1062 Shape tuple_shape = ShapeUtil::MakeTupleShape({shape, shape}); in TEST_P() local
1213 Shape tuple_shape = ShapeUtil::MakeTupleShape({shape, shape}); in TEST_P() local
2586 Shape tuple_shape = ShapeUtil::MakeTupleShape({shape, scalar_shape}); in TEST_P() local
2909 Shape tuple_shape = in TEST_P() local
3001 Shape tuple_shape = ShapeUtil::MakeTupleShape({shape, scalar_shape, shape}); in TEST_P() local
3132 Shape tuple_shape = ShapeUtil::MakeTupleShape({shape, shape}); in TEST_P() local
3176 Shape tuple_shape = ShapeUtil::MakeTupleShape({shape, shape}); in TEST_P() local
3213 Shape tuple_shape = ShapeUtil::MakeTupleShape({shape, shape}); in TEST_P() local
[all …]
Dhlo_sharding_test.cc413 auto tuple_shape = in TEST_P() local
424 auto tuple_shape = ShapeUtil::MakeTupleShape( in TEST_P() local
Dhlo_matchers_test.cc169 auto tuple_shape = ShapeUtil::MakeTupleShape( in TEST_F() local
Dshaped_buffer_test.cc159 Shape tuple_shape = in TEST() local
Dcopy_insertion_test.cc1422 const Shape tuple_shape = in TEST_F() local
1455 const Shape tuple_shape = in TEST_F() local
1488 const Shape tuple_shape = in TEST_F() local
1521 const Shape tuple_shape = in TEST_F() local
1556 const Shape tuple_shape = in TEST_F() local
1594 const Shape tuple_shape = in TEST_F() local
Dhlo_sharding.cc257 HloSharding HloSharding::Tuple(const Shape& tuple_shape, in Tuple()
271 HloSharding HloSharding::SingleTuple(const Shape& tuple_shape, in SingleTuple()
Dhlo_live_range_test.cc267 Shape tuple_shape = ShapeUtil::MakeTupleShape({shape, scalar_shape}); in TEST_F() local
Dhlo_dataflow_analysis_test.cc396 const Shape tuple_shape = in TEST_P() local
499 const Shape tuple_shape = in TEST_P() local
574 const Shape tuple_shape = ShapeUtil::MakeTupleShape({scalar_shape_}); in TEST_P() local
667 const Shape tuple_shape = in TEST_P() local
783 const Shape tuple_shape = in TEST_P() local
836 const Shape tuple_shape = in TEST_P() local
Dconditional_simplifier.cc104 Shape tuple_shape = ShapeUtil::MakeTupleShapeWithPtrs(new_tuple_shapes); in TryRemoveUnusedConditionalOperands() local
Dshape_inference_test.cc1370 Shape tuple_shape = ShapeUtil::MakeTupleShape({f32_, s32_}); in TEST_F() local
1382 Shape tuple_shape = ShapeUtil::MakeTupleShape({f32_, s32_}); in TEST_F() local
2160 Shape tuple_shape = ShapeUtil::MakeTupleShape({input_shape, input_shape}); in TEST_F() local
3342 Shape tuple_shape = in TEST_P() local
3360 Shape tuple_shape = in TEST_P() local
3378 Shape tuple_shape = in TEST_P() local
Dtriangular_solve_expander.cc302 Shape tuple_shape = ShapeUtil::MakeTupleShape(tuple_shapes); in InvertDiagonalBlocks() 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.cc82 const Shape& tuple_shape) { in CreateTupleCeilComputation()
104 const std::string& computation_name, const Shape& tuple_shape) { in CreateTupleFloorComputation()
126 const Shape& tuple_shape) { in CreateTupleAddComputation()
146 const Shape& tuple_shape) { in CreateTupleSubComputation()
636 Shape tuple_shape = ShapeUtil::MakeTupleShape({r0bool, r0f32_, r0f32_}); in XLA_TEST_F() local
666 Shape tuple_shape = ShapeUtil::MakeTupleShape({r0bool, r0f32_, r0f32_}); in XLA_TEST_F() local
706 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
460 const Shape tuple_shape = ShapeUtil::MakeTupleShape(inner_tuple_shapes); in XLA_TEST_F() local
Dparams_test.cc443 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.cc63 const Shape& tuple_shape, llvm::IRBuilder<>* b) { in EmitTupleAllocasAtFunctionEntry()
/external/tensorflow/tensorflow/compiler/xla/
Dshape_util_test.cc461 Shape tuple_shape = in TEST() local
496 Shape tuple_shape = ShapeUtil::MakeTupleShape({array_shape, array_shape}); in TEST() local
751 Shape tuple_shape = ShapeUtil::MakeTupleShape({shape}); in TEST() local
/external/tensorflow/tensorflow/compiler/xla/python/tpu_driver/client/
Dtpu_client.cc185 std::vector<BorrowingLiteral> leaves, const Shape& tuple_shape, in FromLiterals()
264 Shape tuple_shape = ShapeUtil::MakeTupleShape(child_shapes); in MakeTuple() local
/external/tensorflow/tensorflow/core/tpu/
Dtpu_execute.cc142 const xla::Shape& tuple_shape, in FixTupleTableAsync()
/external/tensorflow/tensorflow/compiler/xla/tools/
Ddriver.cc489 TupleShape tuple_shape = assignment.buffers_shape[allocation_idx]; in main() local

12