Home
last modified time | relevance | path

Searched refs:offset_data (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2xla/
Dfused_batchnorm_reserve_space_test.cc85 Tensor offset_data(DT_FLOAT, TensorShape({10})); in TEST() local
86 FillZeros(&offset_data); in TEST()
88 Const(root.WithOpName("offset"), Input::Initializer(offset_data)); in TEST()
/external/tensorflow/tensorflow/core/kernels/
Dconv_ops_fused_impl.h188 const T* offset_data, const T* estimated_mean_data) in FusedBatchNormOutputKernel()
191 offset_data(offset_data), in FusedBatchNormOutputKernel()
202 const T* offset_base = offset_data + i; in operator()
224 const T* offset_data; member
351 fused_batch_norm.offset_data,
362 fused_batch_norm.offset_data,
376 const T* offset_data = nullptr;
427 InitDataPtr(offset, &args->offset_data);
Dconv_ops_test.cc544 const Tensor& scale_data, const Tensor& offset_data,
651 const Tensor& offset_data, in RunConv2DWithBatchNorm() argument
669 ops::Const(root.WithOpName("offset"), Input::Initializer(offset_data)), in RunConv2DWithBatchNorm()
679 const Tensor& scale_data, const Tensor& offset_data, in RunConv2DWithBatchNormAndRelu() argument
696 ops::Const(root.WithOpName("offset"), Input::Initializer(offset_data)), in RunConv2DWithBatchNormAndRelu()
893 const Tensor& scale_data, const Tensor& offset_data, in VerifyConv2DWithBatchNorm()
897 offset_data, mean_data, variance_data, out); in VerifyConv2DWithBatchNorm()
902 const Tensor& scale_data, const Tensor& offset_data, in VerifyConv2DWithBatchNorm() argument
906 {scale_data, offset_data, mean_data, variance_data}, in VerifyConv2DWithBatchNorm()
923 const Tensor& scale_data, const Tensor& offset_data, in VerifyConv2DWithBatchNormAndRelu() argument
[all …]
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.h550 const DeviceMemory<float>& offset_data,
561 const DeviceMemory<float>& offset_data,
572 const DeviceMemory<float>& offset_data, double epsilon,
583 const DeviceMemory<float>& offset_data, double epsilon,
596 const DeviceMemory<float>& offset_data,
612 const DeviceMemory<float>& offset_data,
724 const DeviceMemory<U>& scale_data, const DeviceMemory<U>& offset_data,
735 const DeviceMemory<U>& scale_data, const DeviceMemory<U>& offset_data,
751 const DeviceMemory<U>& scale_data, const DeviceMemory<U>& offset_data,
Drocm_dnn.cc4134 const DeviceMemory<U>& scale_data, const DeviceMemory<U>& offset_data, in DoFusedBatchNormActivationInferenceImpl() argument
4169 scale_data.opaque(), offset_data.opaque(), mean_data.opaque(), in DoFusedBatchNormActivationInferenceImpl()
4210 const DeviceMemory<float>& offset_data, in DoFusedBatchNormActivationInference() argument
4217 scale_offset_mean_variance_descriptor, scale_data, offset_data, mean_data, in DoFusedBatchNormActivationInference()
4226 const DeviceMemory<float>& offset_data, in DoFusedBatchNormActivationInference() argument
4233 scale_offset_mean_variance_descriptor, scale_data, offset_data, mean_data, in DoFusedBatchNormActivationInference()
4243 const DeviceMemory<U>& scale_data, const DeviceMemory<U>& offset_data, in DoFusedBatchNormActivationForwardImpl() argument
4280 scale_data.opaque(), offset_data.opaque(), batch_mean_data->opaque(), in DoFusedBatchNormActivationForwardImpl()
4322 const DeviceMemory<float>& offset_data, double epsilon, in DoFusedBatchNormActivationForward() argument
4329 scale_offset_mean_variance_descriptor, scale_data, offset_data, epsilon, in DoFusedBatchNormActivationForward()
[all …]
/external/python/cpython3/Lib/
Dtarfile.py717 fileobj = _FileInFile(tarfile.fileobj, tarinfo.offset_data,
758 self.offset_data = 0 # the file's data starts here
1127 self.offset_data = tarfile.fileobj.tell()
1128 offset = self.offset_data
1185 self.offset_data = tarfile.fileobj.tell()
1186 tarfile.offset = self.offset_data + self._block(self.size)
1285 offset = next.offset_data
1322 next.offset_data = tarfile.fileobj.tell()
2153 source.seek(tarinfo.offset_data)
/external/tensorflow/tensorflow/compiler/xla/tests/
Dbatch_normalization_test.cc534 std::unique_ptr<GlobalData> offset_data = in XLA_TEST_P() local
546 {input_data.get(), scale_data.get(), offset_data.get()}, in XLA_TEST_P()
635 std::unique_ptr<GlobalData> offset_data = in XLA_TEST_P() local
653 {input_data.get(), scale_data.get(), offset_data.get(), mean_data.get(), in XLA_TEST_P()
/external/tensorflow/tensorflow/stream_executor/
Ddnn.h2397 const DeviceMemory<float>& offset_data,
2410 const DeviceMemory<float>& offset_data,
2451 const DeviceMemory<float>& offset_data, double epsilon,
2464 const DeviceMemory<float>& offset_data, double epsilon,
2511 const DeviceMemory<float>& offset_data,
2529 const DeviceMemory<float>& offset_data,
/external/python/cpython2/Lib/
Dtarfile.py801 tarinfo.offset_data,
951 self.offset_data = 0 # the file's data starts here
1284 self.offset_data = tarfile.fileobj.tell()
1285 offset = self.offset_data
1369 self.offset_data = tarfile.fileobj.tell()
1370 tarfile.offset = self.offset_data + self._block(self.size)
1426 offset = next.offset_data