Searched refs:MatrixDiagPart (Results 1 – 20 of 20) sorted by relevance
| /external/tensorflow/tensorflow/core/api_def/python_api/ |
| D | api_def_MatrixDiagPart.pbtxt | 2 graph_op_name: "MatrixDiagPart"
|
| /external/tensorflow/tensorflow/core/api_def/java_api/ |
| D | api_def_MatrixDiagPart.pbtxt | 2 graph_op_name: "MatrixDiagPart"
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/ |
| D | MatrixDiagPart.pbtxt | 2 name: "MatrixDiagPart"
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/ |
| D | MatrixDiagPart.pbtxt | 2 name: "MatrixDiagPart"
|
| /external/tensorflow/tensorflow/core/kernels/linalg/ |
| D | matrix_diag_op_gpu.cu.cc | 134 struct MatrixDiagPart<GPUDevice, T> { struct
164 template struct MatrixDiagPart<GPUDevice, T>;
|
| D | matrix_diag_op.h | 48 struct MatrixDiagPart { struct
|
| D | matrix_diag_op.cc | 137 functor::MatrixDiagPart<Device, T>::Compute( in Compute()
386 struct MatrixDiagPart<CPUDevice, T> { struct
463 void MatrixDiagPart<GPUDevice, T>::Compute( \
470 extern template struct MatrixDiagPart<GPUDevice, T>;
|
| /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 | 169 grad_outputs->push_back(MatrixDiagPart(scope, grad_inputs[0])); in MatrixDiagGrad()
|
| /external/tensorflow/tensorflow/lite/g3doc/guide/ |
| D | op_select_allowlist.md | 249 * `raw_ops.MatrixDiagPart`
|
| /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 | 2277 TEST_F(OpTest, MatrixDiagPart) { in TEST_F() argument
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.raw_ops.pbtxt | 2384 name: "MatrixDiagPart"
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.raw_ops.pbtxt | 2384 name: "MatrixDiagPart"
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/tests/ |
| D | legalize-tf.mlir | 562 // MatrixDiagPart
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | ops.pbtxt | 3848 explanation: "Use MatrixDiagPart"
23584 name: "MatrixDiagPart"
|
| /external/tensorflow/tensorflow/core/ops/compat/ |
| D | ops_history.v0.pbtxt | 16063 name: "MatrixDiagPart"
|
| D | ops_history.v2.pbtxt | 34270 name: "MatrixDiagPart"
|
| /external/tensorflow/tensorflow/go/op/ |
| D | wrappers.go | 2388 func MatrixDiagPart(scope *Scope, input tf.Output) (diagonal tf.Output) { func
|