Searched refs:DiagPart (Results 1 – 23 of 23) sorted by relevance
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/ |
| D | DiagPart.pbtxt | 2 name: "DiagPart"
27 name: "DiagPart"
53 name: "DiagPart"
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/ |
| D | DiagPart.pbtxt | 2 name: "DiagPart"
27 name: "DiagPart"
53 name: "DiagPart"
|
| /external/tensorflow/tensorflow/core/api_def/python_api/ |
| D | api_def_DiagPart.pbtxt | 2 graph_op_name: "DiagPart"
|
| /external/tensorflow/tensorflow/core/api_def/java_api/ |
| D | api_def_MatrixDiagPart.pbtxt | 4 name: "linalg.DiagPart"
|
| D | api_def_DiagPart.pbtxt | 2 graph_op_name: "DiagPart"
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_DiagPart.pbtxt | 2 graph_op_name: "DiagPart"
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | diag_op_test.cc | 29 test::graph::DiagPart(g, out, type); in Diag()
|
| /external/tensorflow/tensorflow/core/graph/ |
| D | testlib.h | 215 Node* DiagPart(Graph* g, Node* in, DataType type);
|
| D | testlib.cc | 491 Node* DiagPart(Graph* g, Node* in, DataType type) { in DiagPart() function
|
| /external/tensorflow/tensorflow/cc/gradients/ |
| D | array_grad_test.cc | 130 auto y = DiagPart(scope_, x); in TEST_F()
|
| D | array_grad.cc | 153 grad_outputs->push_back(DiagPart(scope, grad_inputs[0])); in DiagGrad()
|
| /external/tensorflow/tensorflow/lite/g3doc/guide/ |
| D | op_select_allowlist.md | 147 * `raw_ops.DiagPart`
|
| /external/tensorflow/tensorflow/compiler/tf2xla/g3doc/ |
| D | gpu_supported_ops.md | 68 `DiagPart` | `T={complex64,double,float,int32,int64}`
|
| D | cpu_supported_ops.md | 68 `DiagPart` | `T={complex64,double,float,int32,int64}`
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | randomized_tests.cc | 1869 TEST_F(OpTest, DiagPart) { in TEST_F() argument
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/tests/ |
| D | legalize-tf.mlir | 539 // DiagPart
557 %0 = "tf.DiagPart"(%arg0) : (tensor<4x3x4x3xf32>) -> tensor<4x3xf32>
|
| /external/tensorflow/tensorflow/core/ops/compat/ |
| D | ops_history.v0.pbtxt | 10326 name: "DiagPart"
10349 name: "DiagPart"
10373 name: "DiagPart"
|
| D | ops_history.v2.pbtxt | 20672 name: "DiagPart"
20697 name: "DiagPart"
20723 name: "DiagPart"
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.raw_ops.pbtxt | 1228 name: "DiagPart"
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.raw_ops.pbtxt | 1228 name: "DiagPart"
|
| /external/tensorflow/tensorflow/go/op/ |
| D | wrappers.go | 5258 func DiagPart(scope *Scope, input tf.Output) (diagonal tf.Output) { func
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_generated_ops.td | 3370 def TF_DiagPartOp : TF_Op<"DiagPart", [NoSideEffect]> {
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | ops.pbtxt | 13064 name: "DiagPart"
|