Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 7 of 7 for zeros (0.14 sec)

  1. tensorflow/c/eager/tape.h

        return absl::OkStatus();
      }
    
      // We may need to allocate zero inputs for trainable dtypes we don't have JVPs
      // for. Make sure they get cleaned up.
      std::vector<Gradient*> new_zeros;
      auto delete_new_zeros = gtl::MakeCleanup([&new_zeros, this] {
        for (Gradient* tensor : new_zeros) {
          this->vspace_.DeleteGradient(tensor);
        }
      });
      std::vector<Gradient*> in_grads;
    C
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Tue Apr 02 12:40:29 GMT 2024
    - 47.2K bytes
    - Viewed (1)
  2. tensorflow/c/eager/gradients.cc

        }
      }
    
      TF_RETURN_IF_ERROR(GradientTape::ComputeGradient(
          vspace, target_tensor_ids, source_tensor_ids, sources_that_are_targets,
          output_gradients, result, /*build_default_zeros_grads*/ false));
      return absl::OkStatus();
    }
    
    // Helper functions which delegate to `AbstractOperation`, update
    // the state of the ForwardOperation and call the tape as appropriate.
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Thu Feb 15 09:49:45 GMT 2024
    - 19.3K bytes
    - Viewed (0)
  3. tensorflow/c/eager/dlpack.cc

      void* data = dl_tensor->data;
    
      if (dl_tensor->byte_offset != 0) {
        status->status = tensorflow::errors::InvalidArgument(
            "Unsupported byte_offset (", dl_tensor->byte_offset,
            ") from DLPack, must be zero");
        return nullptr;
      }
    
      size_t total_bytes = dl_tensor->dtype.bits / 8;
      for (int i = 0; i < num_dims; i++) {
        total_bytes *= dims[i];
      }
    
      if (dl_tensor->strides != nullptr &&
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Thu Feb 15 09:49:45 GMT 2024
    - 12.8K bytes
    - Viewed (0)
  4. tensorflow/c/c_api_experimental.h

        unsigned char should_enable);
    
    // Create a serialized tensorflow.ConfigProto proto, where:
    //
    // a) ConfigProto.optimizer_options.global_jit_level is set to ON_1 if
    // `enable_xla_compilation` is non-zero, and OFF otherwise.
    // b) ConfigProto.gpu_options.allow_growth is set to `gpu_memory_allow_growth`.
    // c) ConfigProto.device_count is set to `num_cpu_devices`.
    TF_CAPI_EXPORT extern TF_Buffer* TF_CreateConfig(
    C
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Thu Apr 27 21:07:00 GMT 2023
    - 15.1K bytes
    - Viewed (0)
  5. tensorflow/c/eager/c_api_cluster_test.cc

      TF_DeleteTensor(t);
    
      for (int i = 0; i < expected_values.size(); i++) {
        EXPECT_EQ(expected_values[i], actual_values[i])
            << "Mismatch in expected values at (zero-based) index " << i;
      }
    }
    
    void CheckRemoteMatMulExecutesOK(TFE_Context* ctx,
                                     const char* remote_device_name,
                                     const char* local_device_name) {
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Fri Apr 14 10:03:59 GMT 2023
    - 19.3K bytes
    - Viewed (0)
  6. ci/official/containers/linux_arm64/builder.devtoolset/stringop_trunc.patch

    -	}
    -      n = n - (s1 - s) - 1;
    -      if (n == 0)
    -	return s;
    -      goto zero_fill;
    -    }
    -
    - last_chars:
    -  n &= 3;
    -  if (n == 0)
    -    return s;
    -
    -  do
    -    {
    -      c = *s2++;
    -      *++s1 = c;
    -      if (--n == 0)
    -	return s;
    -    }
    -  while (c != '\0');
    -
    - zero_fill:
    -  do
    -    *++s1 = '\0';
    -  while (--n > 0);
    -
    -  return s;
    Others
    - Registered: Tue May 07 12:40:20 GMT 2024
    - Last Modified: Mon Sep 18 14:52:45 GMT 2023
    - 42.9K bytes
    - Viewed (1)
  7. tensorflow/c/eager/parallel_device/parallel_device.cc

      return result;
    }
    
    // Used as an argument to TFE_NewCustomDeviceTensorHandle, indicating how
    // ParallelTensors wrapped in TFE_TensorHandles should be cleaned up once their
    // reference counts drop to zero.
    void ParallelTensorDeallocator(void* data) {
      delete reinterpret_cast<ParallelTensor*>(data);
    }
    
    // Used as an argument to TFE_NewCustomDeviceTensorHandle, for computing the
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Wed Mar 29 22:05:31 GMT 2023
    - 18.3K bytes
    - Viewed (0)
Back to top