Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 1 of 1 for Reset (0.11 sec)

  1. tensorflow/c/eager/gradient_checker.cc

        Range(&vals, 0, num_dims_out);
        AbstractTensorHandle* sum_dims_raw = nullptr;
        TF_RETURN_IF_ERROR(TestTensorHandleWithDims<int32_t, TF_INT32>(
            ctx, vals.data(), vals_shape, 1, &sum_dims_raw));
        sum_dims.reset(sum_dims_raw);
      }
    
      // Reduce sum the output on all dimensions.
      TF_RETURN_IF_ERROR(ops::Sum(ctx, model_out.get(), sum_dims.get(), &outputs[0],
                                  /*keep_dims=*/false, "sum_output"));
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Thu Feb 15 09:49:45 GMT 2024
    - 7.3K bytes
    - Viewed (0)
Back to top