Home
last modified time | relevance | path

Searched refs:xla_input_shapes (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dwhile_op.cc156 xla::Shape body_input_shape = body.xla_input_shapes[0]; in VerifyBodyInputAndOutputShapeMatch()
181 xla::Shape cond_input_shape = cond.xla_input_shapes[0]; in BuildWrappedCond()
200 xla::Shape body_input_shape = body.xla_input_shapes[0]; in BuildWrappedBody()
417 OP_REQUIRES(ctx, body.xla_input_shapes.size() == 1, in Compile()
419 xla::Shape body_input_shape = body.xla_input_shapes[0]; in Compile()
422 OP_REQUIRES(ctx, cond.xla_input_shapes.size() == 1, in Compile()
424 xla::Shape cond_input_shape = cond.xla_input_shapes[0]; in Compile()
Dif_op.cc150 OP_REQUIRES(ctx, then_result.xla_input_shapes.size() == 1, in Compile()
152 xla::Shape then_input_shape = then_result.xla_input_shapes[0]; in Compile()
155 OP_REQUIRES(ctx, else_result.xla_input_shapes.size() == 1, in Compile()
157 xla::Shape else_input_shape = else_result.xla_input_shapes[0]; in Compile()
Dcase_op.cc153 OP_REQUIRES(ctx, branch_results[j].xla_input_shapes.size() == 1, in Compile()
155 xla::Shape branch_input_shape = branch_results[j].xla_input_shapes[0]; in Compile()
161 OP_REQUIRES(ctx, branch_results[j].xla_input_shapes.size() == 1, in Compile()
/external/tensorflow/tensorflow/compiler/jit/
Dxla_compilation_cache.cc147 result.xla_input_shapes.size()); in BuildExecutable()
148 for (int i = 0; i < result.xla_input_shapes.size(); ++i) { in BuildExecutable()
149 argument_layouts[i] = &result.xla_input_shapes[i]; in BuildExecutable()
Dxla_launch_util.cc214 arg_buffers_.reserve(kernel->xla_input_shapes.size() + 1); in PopulateInputs()
215 arg_buffers_.resize(kernel->xla_input_shapes.size()); in PopulateInputs()
220 for (int i = 0; i < kernel->xla_input_shapes.size(); ++i) { in PopulateInputs()
223 const xla::Shape& shape = kernel->xla_input_shapes[i]; in PopulateInputs()
/external/tensorflow/tensorflow/compiler/tf2xla/
Dxla_compiler.h255 std::vector<xla::Shape> xla_input_shapes; member
Dxla_compiler_test.cc1474 EXPECT_EQ(result.xla_input_shapes.size(), 1); in TEST_F()
1492 EXPECT_EQ(result.xla_input_shapes.size(), 2); in TEST_F()
1493 EXPECT_TRUE(result.xla_input_shapes[1].IsToken()); in TEST_F()
Dxla_compiler.cc1092 &arg_expressions, &result->input_mapping, &result->xla_input_shapes, in CompileGraph()