Searched refs:MatrixDiagPart (Results 1 – 17 of 17) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | matrix_diag_op.cc | 75 functor::MatrixDiagPart<Device, T>::Compute( in Compute()
159 struct MatrixDiagPart<CPUDevice, T> { struct
184 void MatrixDiagPart<GPUDevice, T>::Compute( \
187 extern template struct MatrixDiagPart<GPUDevice, T>;
|
| D | matrix_diag_op_gpu.cu.cc | 31 template struct functor::MatrixDiagPart<GPUDevice, T>;
|
| D | matrix_diag_op.h | 71 struct MatrixDiagPart { struct
|
| /external/tensorflow/tensorflow/core/api_def/java_api/ |
| D | api_def_MatrixDiagPart.pbtxt | 2 graph_op_name: "MatrixDiagPart"
|
| /external/tensorflow/tensorflow/core/api_def/python_api/ |
| D | api_def_MatrixDiagPart.pbtxt | 2 graph_op_name: "MatrixDiagPart"
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_MatrixDiagPart.pbtxt | 2 graph_op_name: "MatrixDiagPart"
|
| /external/tensorflow/tensorflow/cc/gradients/ |
| D | array_grad.cc | 160 grad_outputs->push_back(MatrixDiagPart(scope, grad_inputs[0])); in MatrixDiagGrad()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/g3doc/ |
| D | gpu_supported_ops.md | 130 `MatrixDiagPart` | `T={bool,complex64,double,float,int32,int64,uint32,uint64}`
|
| D | cpu_supported_ops.md | 130 `MatrixDiagPart` | `T={bool,complex64,double,float,int32,int64,uint32,uint64}`
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | randomized_tests.cc | 2276 TEST_F(OpTest, MatrixDiagPart) { in TEST_F() argument
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.raw_ops.pbtxt | 1912 name: "MatrixDiagPart"
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.raw_ops.pbtxt | 1912 name: "MatrixDiagPart"
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | ops.pbtxt | 3440 explanation: "Use MatrixDiagPart"
18351 name: "MatrixDiagPart"
|
| /external/tensorflow/tensorflow/go/op/ |
| D | wrappers.go | 2538 func MatrixDiagPart(scope *Scope, input tf.Output) (diagonal tf.Output) { func
|
| /external/tensorflow/tensorflow/core/ops/compat/ |
| D | ops_history.v0.pbtxt | 16063 name: "MatrixDiagPart"
|
| D | ops_history.v1.pbtxt | 35371 name: "MatrixDiagPart"
|
| D | ops_history.v2.pbtxt | 34933 name: "MatrixDiagPart"
|