Home
last modified time | relevance | path

Searched refs:kScalarShape (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dflatten_call_graph_test.cc39 HloInstruction::CreateParameter(0, kScalarShape, "param0")); in MakeScalarComputation()
41 HloInstruction::CreateUnary(kScalarShape, HloOpcode::kNegate, param0)); in MakeScalarComputation()
51 HloInstruction::CreateParameter(0, kScalarShape, "param0")); in MakeMappingComputation()
55 kScalarShape, {last_value}, map_computation)); in MakeMappingComputation()
67 HloInstruction::CreateParameter(0, kScalarShape, "param0")); in MakeCallingComputation()
71 kScalarShape, {last_value}, callee_computation)); in MakeCallingComputation()
81 HloInstruction::CreateParameter(0, kScalarShape, "param0")); in MakeConditionComputation()
96 const Shape kScalarShape = ShapeUtil::MakeShape(F32, {}); member in xla::__anonb6a180e00111::FlattenCallGraphTest
124 HloInstruction::CreateParameter(0, kScalarShape, "param0")); in TEST_F()
126 HloInstruction::CreateCall(kScalarShape, {param0}, c_computation)); in TEST_F()
[all …]
Dcall_graph_test.cc41 HloInstruction::CreateParameter(0, kScalarShape, "param0")); in MakeScalarComputation()
43 HloInstruction::CreateUnary(kScalarShape, opcode, param0)); in MakeScalarComputation()
53 HloInstruction::CreateParameter(0, kScalarShape, "param0")); in MakeMappingComputation()
57 kScalarShape, {last_value}, map_computation)); in MakeMappingComputation()
69 HloInstruction::CreateParameter(0, kScalarShape, "param0")); in MakeCallingComputation()
73 kScalarShape, {last_value}, callee_computation)); in MakeCallingComputation()
83 HloInstruction::CreateParameter(0, kScalarShape, "param0")); in MakeConditionComputation()
92 const Shape kScalarShape = ShapeUtil::MakeShape(F32, {}); member in xla::__anonbba859f30111::CallGraphTest
211 HloInstruction::CreateParameter(0, kScalarShape, "param0")); in TEST_F()
213 HloInstruction::CreateCall(kScalarShape, {param0}, subcomputation)); in TEST_F()
[all …]