Searched refs:op_shape (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | zero_sized_hlo_elimination_test.cc | 86 Shape op_shape = ShapeUtil::MakeShape(F32, {4, 0}); in TEST_F() local
87 op_shape.clear_layout(); in TEST_F()
89 HloInstruction::CreateParameter(1, op_shape, "zero sized param 1")); in TEST_F()
91 HloInstruction::CreateParameter(2, op_shape, "zero sized param 2")); in TEST_F()
93 HloInstruction::CreateBinary(op_shape, HloOpcode::kAdd, param1, param2)); in TEST_F()
|
| /external/tensorflow/tensorflow/contrib/solvers/python/kernel_tests/ |
| D | util_test.py | 48 op_shape = ops.convert_to_tensor(op.shape)
50 op_shape_val, ax_val, aty_val = sess.run([op_shape, ax, aty])
53 [op_shape, ax, aty], feed_dict={a: a_np,
83 op_shape = ops.convert_to_tensor(id_op.shape)
85 op_shape_val, ax_val, aty_val = sess.run([op_shape, ax, aty])
88 [op_shape, ax, aty], feed_dict={
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | xla_expression_test.cc | 101 TF_ASSERT_OK_AND_ASSIGN(TensorShape op_shape, in TEST_F()
103 EXPECT_EQ(TensorShape({}), op_shape); in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | ir_emitter_unnested.cc | 3570 const HloInstruction* unnested_hlo, const Shape& op_shape, in NumInputsInvolveInOnlyElementwiseOps() argument
3575 return ShapeUtil::SameDimensions(op_shape, parameter_shape) && in NumInputsInvolveInOnlyElementwiseOps()
|