- Sort Score
- Result 10 results
- Languages All
Results 1 - 6 of 6 for DiagPart (0.16 sec)
-
tensorflow/cc/framework/fuzzing/cc_op_fuzz_gen.cc
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sat Jan 27 16:26:51 UTC 2024 - 13K bytes - Viewed (0) -
tensorflow/cc/gradients/array_grad.cc
REGISTER_GRADIENT_OP("Fill", FillGrad); Status DiagGrad(const Scope& scope, const Operation& op, const std::vector<Output>& grad_inputs, std::vector<Output>* grad_outputs) { grad_outputs->push_back(DiagPart(scope, grad_inputs[0])); return scope.status(); } REGISTER_GRADIENT_OP("Diag", DiagGrad); Status DiagPartGrad(const Scope& scope, const Operation& op, const std::vector<Output>& grad_inputs,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Oct 10 23:33:32 UTC 2023 - 31.7K bytes - Viewed (0) -
tensorflow/cc/gradients/array_grad_test.cc
RunTest(x, x_shape, y, y_shape); } TEST_F(ArrayGradTest, DiagPartGrad) { TensorShape x_shape({5, 2, 5, 2}); auto x = Placeholder(scope_, DT_FLOAT, Placeholder::Shape(x_shape)); auto y = DiagPart(scope_, x); TensorShape y_shape({5, 2}); RunTest(x, x_shape, y, y_shape); } TEST_F(ArrayGradTest, MatrixDiagGrad) { TensorShape x_shape({5, 2});
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Oct 10 23:33:32 UTC 2023 - 19.3K bytes - Viewed (0) -
tensorflow/compiler/jit/mark_for_compilation_pass.cc
{"All", "Any", "Min", "Max", "Mean", "Prod", "Sum"}}, // clang-format on {"PWRED", {"ArgMax", "ArgMin", "DiagPart", "Softmax", "SparseSoftmaxCrossEntropyWithLogits", "LogSoftmax"}}, {"REDUCEWINDOW", {"ArgMax", "ArgMin", "DiagPart", "Softmax", "SparseSoftmaxCrossEntropyWithLogits", "LogSoftmax"}}, {"REDUCEWINDOWPW", {"BiasAddGrad", "LRN", "LRNGrad"}},
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Feb 21 12:19:41 UTC 2024 - 85.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/tests/legalize-tf.mlir
// CHECK: return [[CLAMP]] func.return %0 : tensor<?xf32> } //===----------------------------------------------------------------------===// // DiagPart //===----------------------------------------------------------------------===// // ----- // CHECK-LABEL: func @diag_part // CHECK-SAME: %[[ARG:.*]]: tensor<4x3x4x3xf32>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon May 06 18:46:23 UTC 2024 - 335.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_generated_ops.td
); TF_DerivedOperandTypeAttr T = TF_DerivedOperandTypeAttr<0>; } def TF_DiagPartOp : TF_Op<"DiagPart", [Pure]> { let summary = "Returns the diagonal part of the tensor."; let description = [{ This operation returns a tensor with the `diagonal` part of the `input`. The `diagonal` part is computed as follows:
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 11 23:24:08 UTC 2024 - 793K bytes - Viewed (0)