- Sort Score
- Result 10 results
- Languages All
Results 11 - 20 of 38 for zeros_like (0.17 sec)
-
tensorflow/compiler/mlir/tf2xla/tests/legalize-tf.mlir
} //===----------------------------------------------------------------------===// // ZerosLike //===----------------------------------------------------------------------===// // ----- // CHECK-LABEL: @zeros_like // CHECK-SAME: (%[[ARG:.*]]: tensor<2x?xf32>) func.func @zeros_like(%arg0: tensor<2x?xf32>) -> tensor<2x?xf32> {
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/cc/gradients/math_grad.cc
auto gathered = Gather(scope, params, zero_clipped_indices); // Replace gathered params of negative indices with 0. auto zero_slice = ZerosLike(scope, gathered); return SelectV2(scope, is_positive, gathered, zero_slice); } Status UnsortedSegmentMinOrMaxGrad(const Scope& scope, const Operation& op, const std::vector<Output>& grad_inputs,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Aug 25 18:20:20 UTC 2023 - 50.7K bytes - Viewed (0) -
tensorflow/c/eager/gradients.h
// gradients. // // `ZerosLike` is not expected to be called and returns a nullptr. The creation // of default zeros grads is handled by the `DefaultGradientFunction` registered // for each op. // TODO(srbs): We need to define `ZerosLike` here to keep the compiler happy. // Figure out a way to avoid this. // TODO(srbs): Should ZerosLike check-fail instead of returning nullptr? class TapeTensor {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Sep 26 10:27:05 UTC 2022 - 6.9K bytes - Viewed (0) -
tensorflow/c/experimental/ops/array_ops.cc
TF_RETURN_IF_ERROR(op_ptr->AddInputList(input)); int num_retvals = output.size(); return op_ptr->Execute(output, &num_retvals); } // Op: ZerosLike() // Summary: Returns a tensor of zeros with the same shape and type as x. // // Description: Status ZerosLike(AbstractContext* ctx, AbstractTensorHandle* const x, AbstractTensorHandle** y, const char* name, const char* raw_device_name) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 10 19:11:36 UTC 2022 - 6.7K bytes - Viewed (0) -
tensorflow/c/experimental/ops/update_cpp_ops.sh
--output_dir="${current_dir}" \ --api_dirs="${api_dir}" \ --source_dir=third_party/tensorflow" ${generate} \ --category=array \ Identity \ IdentityN \ ZerosLike \ Shape \ ExpandDims \ OnesLike ${generate} \ --category=math \ Mul \ Conj \ AddV2 \ MatMul \ Neg \ Sum \ Sub \ Div \ DivNoNan \ Exp \ Sqrt \
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 17 17:54:34 UTC 2022 - 1.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfrt/tests/tf_to_corert/attributes.mlir
func.return } // CHECK-LABEL: func @variant func.func @variant(%arg: tensor<!tf_type.variant>) { // CHECK: tf.ZerosLike // CHECK-SAME: T = !corert.variant %0 = "tf.ZerosLike"(%arg) {device = "/device:CPU:0", T = !tf_type.variant} : (tensor<!tf_type.variant>) -> tensor<!tf_type.variant> func.return } // Checks that TF quantized attrs are lowered to the corert types
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 08 00:18:59 UTC 2024 - 4.8K bytes - Viewed (0) -
tensorflow/c/experimental/ops/array_ops.h
const char* name = nullptr, const char* raw_device_name = nullptr); // Returns a tensor of zeros with the same shape and type as x. Status ZerosLike(AbstractContext* ctx, AbstractTensorHandle* const x, AbstractTensorHandle** y, const char* name = nullptr, const char* raw_device_name = nullptr); // Returns the shape of a tensor.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 10 19:11:36 UTC 2022 - 2.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/ir/tfl_ops.td
}]; let hasOptions = 1; } def TFL_ZerosLikeOp: TFL_Op<"zeros_like", [ PredOpTrait<"input and output must have same element type", TFL_TCresVTEtIsSameAsOp<0, 0>>, SameOperandsAndResultShape, Pure]> { let summary = "ZerosLike operator"; let description = [{ Returns a tensor of zeros with the same shape and type as the input tensor.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jun 06 19:09:08 UTC 2024 - 186K bytes - Viewed (0) -
tensorflow/c/eager/gradients.cc
int64_t ToId(const AbstractTensorHandle* t) { return static_cast<int64_t>(reinterpret_cast<uintptr_t>(t)); } Status ZerosLike(AbstractContext* ctx, AbstractTensorHandle* t, AbstractTensorHandle** result) { AbstractOperationPtr op(ctx->CreateOperation()); TF_RETURN_IF_ERROR(op->Reset("ZerosLike", /*raw_device_name=*/nullptr)); if (isa<tracing::TracingOperation>(op.get())) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 15 09:49:45 UTC 2024 - 19.3K bytes - Viewed (0) -
src/runtime/iface_test.go
{name: "E32", fn: func() { e = zero32 }}, {name: "E64", fn: func() { e = zero64 }}, {name: "Estr", fn: func() { e = zerostr }}, {name: "Eslice", fn: func() { e = zeroslice }}, {name: "Econstflt", fn: func() { e = 99.0 }}, // constants do not allocate {name: "Econststr", fn: func() { e = "change" }}, {name: "I8", fn: func() { i1 = eight8I }}, {name: "I16", fn: func() { i1 = zero16I }},
Registered: Wed Jun 12 16:32:35 UTC 2024 - Last Modified: Mon Dec 13 18:45:54 UTC 2021 - 7.5K bytes - Viewed (0)