Searched refs:DiagPart (Results 1 – 19 of 19) sorted by relevance
| /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/python_api/ |
| 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 | 214 Node* DiagPart(Graph* g, Node* in, DataType type);
|
| D | testlib.cc | 490 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 | 144 grad_outputs->push_back(DiagPart(scope, grad_inputs[0])); in DiagGrad()
|
| /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 | 1868 TEST_F(OpTest, DiagPart) { in TEST_F() argument
|
| /external/tensorflow/tensorflow/core/ops/compat/ |
| D | ops_history.v0.pbtxt | 10326 name: "DiagPart"
10349 name: "DiagPart"
10373 name: "DiagPart"
|
| D | ops_history.v1.pbtxt | 21015 name: "DiagPart"
21040 name: "DiagPart"
21066 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 | 904 name: "DiagPart"
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.raw_ops.pbtxt | 904 name: "DiagPart"
|
| /external/tensorflow/tensorflow/go/op/ |
| D | wrappers.go | 2706 func DiagPart(scope *Scope, input tf.Output) (diagonal tf.Output) { func
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | ops.pbtxt | 9740 name: "DiagPart"
|