Home
last modified time | relevance | path

Searched refs:layout_output (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dxsmm_conv2d.cc139 libxsmm_dnn_tensor_datalayout* layout_output; member
175 libxsmm_dnn_destroy_tensor_datalayout(i->second.layout_output), in ~libxsmm_dnn_registry_type()
211 regentry.layout_output = libxsmm_dnn_create_tensor_datalayout( in find()
347 libxsmm_dnn_link_tensor(regentry.layout_output, output, &status); in CallLibxsmmConvGeneric()
/external/tensorflow/tensorflow/compiler/mlir/hlo/include/mlir-hlo/Dialect/mhlo/IR/
Dlhlo_ops.td641 HLO_LayoutAttr:$layout_output
/external/tensorflow/tensorflow/compiler/mlir/xla/tests/hlo_to_lhlo_with_xla/
Dhlo_text_to_lhlo_no_opt.hlotxt675 // CHECK-SAME: layout_output = dense<[1, 0]> : tensor<2xindex>
/external/tensorflow/tensorflow/compiler/mlir/hlo/tests/
Dlhlo_ops.mlir871 layout_output = dense<[1, 0]> : tensor<2xindex>,
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter_unnested.cc1608 TF_RET_CHECK(has_fortran_layout(triangular_solve_op.layout_output())); in EmitTriangularSolveFromMlir()