Searched refs:ConstantFromArray (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | select_and_scatter_test.cc | 76 auto operand = ConstantFromArray(&builder_, o); in XLA_TEST_P()
81 auto source = ConstantFromArray(&builder_, s); in XLA_TEST_P()
|
| D | dot_operation_test.cc | 116 auto rhs = ConstantFromArray<T>(&builder, {3.0f, 4.0f}); in XLA_TYPED_TEST()
137 auto lhs = ConstantFromArray<T>(&builder, {1.0f, 2.5f, 42.0f}); in XLA_TYPED_TEST()
138 auto rhs = ConstantFromArray<T>(&builder, {11.0f, -1.0f, 0.5f}); in XLA_TYPED_TEST()
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.h | 1595 XlaOp ConstantFromArray(XlaBuilder* builder, const Array<NativeT>& values);
2594 XlaOp ConstantFromArray(XlaBuilder* builder, const Array<NativeT>& values) { in ConstantFromArray() function
2626 return ConstantFromArray(builder, values); in ConstantR3FromArray3D()
2639 return ConstantFromArray(builder, values); in ConstantR4FromArray4D()
|