Searched refs:MatrixSetDiag (Results 1 – 25 of 25) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/linalg/ |
| D | matrix_set_diag_op.cc | 163 functor::MatrixSetDiag<Device, T>::Compute( in Compute()
203 struct MatrixSetDiag<CPUDevice, T> { struct
265 void MatrixSetDiag<GPUDevice, T>::Compute( \
273 extern template struct MatrixSetDiag<GPUDevice, T>;
|
| D | matrix_set_diag_op_gpu.cu.cc | 100 struct MatrixSetDiag<GPUDevice, Scalar> { struct
137 #define DEFINE_GPU_SPEC(T) template struct MatrixSetDiag<GPUDevice, T>;
|
| D | matrix_set_diag_op.h | 27 struct MatrixSetDiag { struct
|
| /external/tensorflow/tensorflow/core/api_def/python_api/ |
| D | api_def_MatrixSetDiag.pbtxt | 2 graph_op_name: "MatrixSetDiag"
|
| /external/tensorflow/tensorflow/core/api_def/java_api/ |
| D | api_def_MatrixSetDiag.pbtxt | 2 graph_op_name: "MatrixSetDiag"
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/ |
| D | MatrixSetDiag.pbtxt | 2 name: "MatrixSetDiag"
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/ |
| D | MatrixSetDiag.pbtxt | 2 name: "MatrixSetDiag"
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_MatrixSetDiag.pbtxt | 2 graph_op_name: "MatrixSetDiag"
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/transforms/ |
| D | prepare_patterns.td | 144 // `MatrixSetDiagV3` to `MatrixSetDiag` with default `k` inputs.
|
| D | legalize_patterns.td | 455 // TFLite only supports MatrixSetDiag ops with scalar zero k attribute.
459 "MatrixSetDiag attribute verification">;
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | canonicalize.td | 172 // MatrixSetDiag op patterns.
|
| /external/tensorflow/tensorflow/lite/g3doc/guide/ |
| D | op_select_allowlist.md | 255 * `raw_ops.MatrixSetDiag`
|
| /external/tensorflow/tensorflow/compiler/tf2xla/g3doc/ |
| D | gpu_supported_ops.md | 131 `MatrixSetDiag` | `T={bool,complex64,double,float,int32,int64,uint32,uint64}`
|
| D | cpu_supported_ops.md | 131 `MatrixSetDiag` | `T={bool,complex64,double,float,int32,int64,uint32,uint64}`
|
| /external/tensorflow/tensorflow/lite/toco/ |
| D | tooling_util.cc | 452 HANDLE_OPERATORTYPENAME_CASE(MatrixSetDiag) in OperatorTypeName()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/ |
| D | canonicalize.mlir | 1371 %0 = "tf.MatrixSetDiag"(%arg0, %arg1) : (tensor<3x3xi64>, tensor<3xi64>) -> tensor<3x3xi64>
|
| /external/tensorflow/tensorflow/compiler/mlir/tosa/g3doc/ |
| D | legalization.md | 2145 ### tf.MatrixSetDiag
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.raw_ops.pbtxt | 2416 name: "MatrixSetDiag"
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.raw_ops.pbtxt | 2416 name: "MatrixSetDiag"
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | ops.pbtxt | 3903 explanation: "Use MatrixSetDiag"
23817 name: "MatrixSetDiag"
|
| /external/tensorflow/ |
| D | RELEASE.md | 2096 Any, and MatrixSetDiag.
|
| /external/tensorflow/tensorflow/core/ops/compat/ |
| D | ops_history.v0.pbtxt | 16127 name: "MatrixSetDiag"
|
| D | ops_history.v2.pbtxt | 34471 name: "MatrixSetDiag"
|
| /external/tensorflow/tensorflow/go/op/ |
| D | wrappers.go | 10552 func MatrixSetDiag(scope *Scope, input tf.Output, diagonal tf.Output) (output tf.Output) { func
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_generated_ops.td | 7665 def TF_MatrixSetDiagOp : TF_Op<"MatrixSetDiag", [NoSideEffect]> {
|