Home
last modified time | relevance | path

Searched refs:InvertPermutation (Results 1 – 22 of 22) sorted by relevance

/external/tensorflow/tensorflow/core/api_def/java_api/
Dapi_def_InvertPermutation.pbtxt2 graph_op_name: "InvertPermutation"
4 name: "math.InvertPermutation"
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dtranspose_op.cc130 InvertPermutation<int32>(ctx); in Compile()
133 InvertPermutation<int64>(ctx); in Compile()
145 void InvertPermutation(XlaOpKernelContext* ctx) { in InvertPermutation() function in tensorflow::__anon5c50effa0111::InvertPermutationOp
/external/tensorflow/tensorflow/core/api_def/python_api/
Dapi_def_InvertPermutation.pbtxt2 graph_op_name: "InvertPermutation"
/external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/
DInvertPermutation.pbtxt2 name: "InvertPermutation"
/external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/
DInvertPermutation.pbtxt2 name: "InvertPermutation"
/external/tensorflow/tensorflow/core/api_def/base_api/
Dapi_def_InvertPermutation.pbtxt2 graph_op_name: "InvertPermutation"
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/
Dlower_tf.mlir9 %0 = "tf.InvertPermutation"(%arg0) : (tensor<5xi32>) -> tensor<5xi32>
15 // CHECK: tf.InvertPermutation
16 %0 = "tf.InvertPermutation"(%arg0) : (tensor<?xi32>) -> tensor<?xi32>
22 // CHECK: tf.InvertPermutation
23 %0 = "tf.InvertPermutation"(%arg0) : (tensor<*xi32>) -> tensor<*xi32>
Dtf-ops.mlir2700 // expected-error @+1 {{'tf.InvertPermutation' op requires input x to be 1-dimensional}}
2701 %0 = "tf.InvertPermutation"(%arg0) : (tensor<8x8xi32>) -> tensor<8x8xi32>
/external/tensorflow/tensorflow/cc/gradients/
Darray_grad.cc244 auto inverted_perm = InvertPermutation(scope, op.input(1)); in TransposeGrad()
Dmath_grad.cc1048 Mul(scope, grad_tiled, Transpose(scope, y, InvertPermutation(scope, perm))); in ProdGrad()
/external/tensorflow/tensorflow/lite/g3doc/guide/
Dop_select_allowlist.md228 * `raw_ops.InvertPermutation`
/external/tensorflow/tensorflow/compiler/tf2xla/g3doc/
Dgpu_supported_ops.md110 `InvertPermutation` | `T={int32}`
Dcpu_supported_ops.md110 `InvertPermutation` | `T={int32}`
/external/tensorflow/tensorflow/compiler/mlir/tosa/g3doc/
Dlegalization.md1954 ### tf.InvertPermutation
/external/tensorflow/tensorflow/compiler/jit/tests/
Dopens2s_gnmt_mixed_precision.pbtxt.gz
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.raw_ops.pbtxt1992 name: "InvertPermutation"
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.raw_ops.pbtxt1992 name: "InvertPermutation"
/external/tensorflow/tensorflow/core/ops/compat/
Dops_history.v0.pbtxt14515 name: "InvertPermutation"
14526 name: "InvertPermutation"
Dops_history.v2.pbtxt30277 name: "InvertPermutation"
/external/tensorflow/tensorflow/go/op/
Dwrappers.go1934 func InvertPermutation(scope *Scope, x tf.Output) (y tf.Output) { func
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
Dtf_generated_ops.td5664 def TF_InvertPermutationOp : TF_Op<"InvertPermutation", [NoSideEffect]> {
/external/tensorflow/tensorflow/core/ops/
Dops.pbtxt20078 name: "InvertPermutation"