| /external/tensorflow/tensorflow/core/framework/ |
| D | resource_op_kernel_test.cc | 90 std::unique_ptr<StubResourceOpKernel> CreateOp(int code, in CreateOp() function in tensorflow::__anon7da9c82a0111::ResourceOpKernelTest
135 auto op = CreateOp(code, ""); in TEST_F()
163 auto op = CreateOp(code, shared_name); in TEST_F()
182 auto op1 = CreateOp(-333, "shared_stub"); in TEST_F()
183 auto op2 = CreateOp(-333, "shared_stub"); in TEST_F()
193 auto op1 = CreateOp(-444, "shared_stub"); in TEST_F()
194 auto op2 = CreateOp(0, "shared_stub"); // Different resource code. in TEST_F()
|
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | mkl_eager_op_rewrite_test.cc | 36 std::unique_ptr<tensorflow::EagerOperation> CreateOp(const string op_name) { in CreateOp() function in tensorflow::EagerOpRewriteTest
89 auto orig_op = CreateOp(conv_ops[i]); \
103 auto orig_op = CreateOp(conv_ops[i]); \
117 auto orig_op = CreateOp(conv_ops[i]); \
141 auto orig_op = CreateOp(ops[i]); \
155 auto orig_op = CreateOp(Fused_BN_ops[i]); \
|
| /external/tensorflow/tensorflow/core/kernels/image/ |
| D | scale_and_translate_op_test.cc | 197 void CreateOp(const string& kernel_type_str, const bool antialias) { in CreateOp() function in tensorflow::ScaleAndTranslateOpTest
272 CreateOp("lanczos3", true); in TEST_F()
288 CreateOp("lanczos3", true); in TEST_F()
304 CreateOp("lanczos3", true); in TEST_F()
320 CreateOp("lanczos3", true); in TEST_F()
337 CreateOp("lanczos3", false); in TEST_F()
354 CreateOp("lanczos3", true); in TEST_F()
370 CreateOp("lanczos3", false); in TEST_F()
389 CreateOp("lanczos3", true); in TEST_F()
405 CreateOp("lanczos3", false); in TEST_F()
[all …]
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/include/mlir-hlo/Dialect/mhlo/transforms/ |
| D | map_chlo_to_hlo_op.h | 29 static mhlo::ComplexOp CreateOp(BroadcastComplexOp from_op, Type result_type, in CreateOp() function
38 static ToOpTy CreateOp(FromOpTy from_op, Type result_type, in CreateOp() function
46 static mhlo::CompareOp CreateOp(BroadcastCompareOp from_op, Type result_type, in CreateOp() function
|
| D | map_lmhlo_to_scalar_op.h | 276 return MapLhloOpToStdScalarOpImpl<complex::CreateOp>{}(loc, result_types,
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/ir/ |
| D | mlir_hlo_builder.cc | 350 return CreateOp(op_name, shape, operands); in RngOpInternal()
425 return CreateOp(GetMlirOpName(binop), shape, {lhs, rhs}); in BinaryOpNoBroadcast()
431 return CreateOp(GetMlirOpName(opcode), shape, in AddOpWithShape()
568 StatusOr<XlaOp> MlirHloBuilder::CreateOp( in CreateOp() function in xla::MlirHloBuilder
|
| D | mlir_hlo_builder.h | 262 StatusOr<XlaOp> CreateOp(
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/ |
| D | transform_unranked_hlo.cc | 292 Adaptor::CreateOp(op, result_type, lhs, rhs, if_eq_shapes_builder); in matchAndRewrite()
|
| D | chlo_legalize_to_hlo.cc | 1029 op, {Adaptor::CreateOp(op, op.getResult().getType(), operands[0], in matchAndRewrite()
1123 Value final_result = Adaptor::CreateOp(op, result_type, broadcasted_lhs, in matchAndRewrite()
|