Home
last modified time | relevance | path

Searched refs:dim_index (Results 1 – 25 of 45) sorted by relevance

12

/external/tensorflow/tensorflow/core/kernels/
Dreduction_ops_common.cc111 int dim_index = 0; in Simplify() local
112 for (; dim_index < data.dims(); ++dim_index) { in Simplify()
113 if (data.dim_size(dim_index) != 1) break; in Simplify()
115 if (dim_index >= data.dims()) { in Simplify()
129 reduce_first_axis_ = bitmap[dim_index]; in Simplify()
130 data_reshape_.push_back(data.dim_size(dim_index)); in Simplify()
131 ++dim_index; in Simplify()
132 for (; dim_index < data.dims(); ++dim_index) { in Simplify()
133 const auto size = data.dim_size(dim_index); in Simplify()
135 bitmap[dim_index] = bitmap[dim_index - 1]; in Simplify()
[all …]
/external/eigen/unsupported/Eigen/CXX11/src/Tensor/
DTensorPadding.h205 Index index, int dim_index) const {
207 return (!internal::index_pair_first_statically_eq<PaddingDimensions>(dim_index, 0) &&
208 index < m_padding[dim_index].first) ||
209 (!internal::index_pair_second_statically_eq<PaddingDimensions>(dim_index, 0) &&
210 index >= m_dimensions[dim_index] - m_padding[dim_index].second);
212 return (index < m_padding[dim_index].first) ||
213 (index >= m_dimensions[dim_index] - m_padding[dim_index].second);
218 int dim_index) const {
220 return internal::index_pair_first_statically_eq<PaddingDimensions>(dim_index, 0);
222 EIGEN_UNUSED_VARIABLE(dim_index);
[all …]
/external/skia/modules/skottie/src/effects/
DGaussianBlurEffect.cpp57 const auto dim_index = SkTPin<size_t>(static_cast<size_t>(fDimensions), in onSync() local
62 fBlur->setSigma({ sigma * kDimensionsMap[dim_index].x(), in onSync()
63 sigma * kDimensionsMap[dim_index].y() }); in onSync()
/external/tensorflow/tensorflow/compiler/xla/service/
Ddynamic_padder.cc759 for (int64 dim_index = 0; dim_index < input_window->dimensions_size(); in RewriteInputWithDynamicPadding() local
760 ++dim_index) { in RewriteInputWithDynamicPadding()
761 if (padding_before[dim_index] == nullptr) { in RewriteInputWithDynamicPadding()
764 int64 shape_dim = window_dim_to_shape_dim(dim_index); in RewriteInputWithDynamicPadding()
766 WindowDimension* window_dim = input_window->mutable_dimensions(dim_index); in RewriteInputWithDynamicPadding()
784 padding_before[dim_index])); in RewriteInputWithDynamicPadding()
1031 for (int64 dim_index = 0; dim_index < rank; ++dim_index) { in RewriteDynamicReduceWindowSamePadding() local
1034 dim_index); in RewriteDynamicReduceWindowSamePadding()
1038 const WindowDimension& window_dim = window.dimensions(dim_index); in RewriteDynamicReduceWindowSamePadding()
1042 input = PadWithScalar(input, dim_index, operand_dynamic_size, init); in RewriteDynamicReduceWindowSamePadding()
[all …]
/external/tensorflow/tensorflow/core/framework/
Dtensor.cc1017 void PrintOneDim(int dim_index, const gtl::InlinedVector<int64, 4>& shape, in PrintOneDim() argument
1021 int64 element_count = shape[dim_index]; in PrintOneDim()
1023 if (dim_index == shape_size - 1) { in PrintOneDim()
1027 if (dim_index != 0) { in PrintOneDim()
1045 PrintOneDim(dim_index + 1, shape, limit, shape_size, data, data_index, in PrintOneDim()
1055 void PrintDimSpacing(int dim_index, int num_dims, string* result) { in PrintDimSpacing() argument
1056 if (dim_index == num_dims - 1) { in PrintDimSpacing()
1060 for (int j = 0; j < num_dims - dim_index - 1; j++) { in PrintDimSpacing()
1063 for (int j = 0; j <= dim_index; j++) { in PrintDimSpacing()
1070 void PrintOneDimV2(int dim_index, const gtl::InlinedVector<int64, 4>& shape, in PrintOneDimV2() argument
[all …]
/external/tensorflow/tensorflow/c/eager/
Dc_api_debug.cc83 TFE_TensorDebugInfo* debug_info, int dim_index) { in TFE_TensorDebugInfoOnDeviceDim() argument
84 return debug_info->dev_dims[dim_index]; in TFE_TensorDebugInfoOnDeviceDim()
Dc_api.cc310 int64_t TFE_TensorHandleDim(TFE_TensorHandle* h, int dim_index, in TFE_TensorHandleDim() argument
318 status->status = tensorflow::unwrap(h)->Dim(dim_index, &dim); in TFE_TensorHandleDim()
490 using DimCallback = std::function<int64_t(int dim_index, TF_Status* status)>;
512 Status Dim(int dim_index, int64* dim) const override { in Dim() argument
514 *dim = dim_callback_(dim_index, &s); in Dim()
531 int64_t (*dim_callback)(void* data, int dim_index, void* arg, in TFE_NewCustomDeviceTensorHandle() argument
551 [dim_callback, data, arg](int dim_index, TF_Status* status) { in TFE_NewCustomDeviceTensorHandle() argument
552 return dim_callback(data, dim_index, arg, status); in TFE_NewCustomDeviceTensorHandle()
589 [dimvec](int dim_index, TF_Status* status) { in TFE_NewTensorHandleFromDeviceMemory() argument
590 return dimvec[dim_index]; in TFE_NewTensorHandleFromDeviceMemory()
Dimmediate_execution_tensor_handle.h41 virtual Status Dim(int dim_index, int64* dim) const = 0;
Dc_api.h152 int dim_index,
233 TFE_TensorDebugInfo* debug_info, int dim_index);
/external/tensorflow/tensorflow/core/common_runtime/eager/
Dtensor_handle_data.cc58 Status LocalTensorHandleData::Dim(int dim_index, int64* dim) const { in Dim() argument
61 *dim = tensor_.dim_size(dim_index); in Dim()
Dcustom_device_test.cc76 Status Dim(int dim_index, int64* dim) const override { in Dim() argument
77 if (dim_index == 0) { in Dim()
Dtensor_handle.cc91 Status TensorHandle::PackedTensorHandleData::Dim(int dim_index, in Dim() argument
93 *dim = shape_.dim_size(dim_index); in Dim()
638 Status TensorHandle::Dim(int dim_index, int64* dim) const { in Dim() argument
641 inference_shape_.dim_size(dim_index) != -1) { in Dim()
642 *dim = inference_shape_.dim_size(dim_index); in Dim()
646 [dim_index, dim](auto& data) { return data.Dim(dim_index, dim); }, in Dim()
Dtensor_handle_data.h39 Status Dim(int dim_index, int64* dim) const;
/external/tensorflow/tensorflow/c/
Dtf_tensor.cc153 int64_t TF_Dim(const TF_Tensor* t, int dim_index) { in TF_Dim() argument
154 return t->tensor->Dim(dim_index); in TF_Dim()
203 int64_t TensorInterface::Dim(int dim_index) const { in Dim()
204 return static_cast<int64_t>(tensor_.dim_size(dim_index)); in Dim()
Dtensor_interface.h40 virtual int64_t Dim(int dim_index) const = 0;
Dtf_tensor.h121 TF_CAPI_EXPORT extern int64_t TF_Dim(const TF_Tensor* tensor, int dim_index);
/external/tensorflow/tensorflow/java/src/main/native/
Deager_operation_jni.cc135 jint dim_index) { in Java_org_tensorflow_EagerOperation_dim() argument
139 int64_t dim = TFE_TensorHandleDim(tensor_handle, dim_index, status); in Java_org_tensorflow_EagerOperation_dim()
/external/tensorflow/tensorflow/lite/python/optimize/
Dcalibration_wrapper.cc190 for (size_t dim_index = 0; dim_index < PyList_Size(shape); ++dim_index) { in Prepare() local
191 PyObject* dim = PyList_GetItem(shape, dim_index); in Prepare()
/external/tensorflow/tensorflow/compiler/xla/
Dliteral.cc229 int32 LiteralBase::GetDynamicSize(int64 dim_index) const { in GetDynamicSize()
230 return GetDynamicSize(dim_index, {}); in GetDynamicSize()
233 int32 LiteralBase::GetDynamicSize(int64 dim_index, in GetDynamicSize() argument
235 return piece(shape_index).GetDynamicSize(dim_index); in GetDynamicSize()
452 int32 LiteralBase::Piece::GetDynamicSize(int64 dim_index) const { in GetDynamicSize()
454 if (!subshape_->is_dynamic_dimension(dim_index)) { in GetDynamicSize()
456 return subshape_->dimensions(dim_index); in GetDynamicSize()
459 return dynamic_size_buffer_[dim_index]; in GetDynamicSize()
462 void LiteralBase::Piece::SetDynamicSize(int64 dim_index, int32 size) { in SetDynamicSize() argument
464 CHECK(subshape_->is_dynamic_dimension(dim_index)); in SetDynamicSize()
[all …]
Dliteral.h124 int32 GetDynamicSize(int64 dim_index, const ShapeIndex& shape_index) const;
125 int32 GetDynamicSize(int64 dim_index) const;
381 int32 GetDynamicSize(int64 dim_index) const;
382 void SetDynamicSize(int64 dim_index, int32 size);
602 void SetDynamicSize(int64 dim_index, const ShapeIndex& shape_index,
604 void SetDynamicSize(int64 dim_index, int32 size);
/external/tensorflow/tensorflow/core/distributed_runtime/eager/
Dremote_tensor_handle_data.cc142 Status RemoteTensorHandleData::Dim(int dim_index, int64* dim) const { in Dim() argument
146 *dim = shape_.dim_size(dim_index); in Dim()
Dremote_tensor_handle_data.h47 Status Dim(int dim_index, int64* dim) const;
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dshape_op.cc132 int64 dim_index; in Compile() local
133 OP_REQUIRES_OK(ctx, ctx->ConstantInputAsIntScalar("dim_index", &dim_index)); in Compile()
136 xla::SetDimensionSize(ctx->Input(0), ctx->Input("size"), dim_index); in Compile()
/external/tensorflow/tensorflow/lite/c/
Dc_api.cc192 int32_t TfLiteTensorDim(const TfLiteTensor* tensor, int32_t dim_index) { in TfLiteTensorDim() argument
193 return tensor->dims->data[dim_index]; in TfLiteTensorDim()
Dc_api.h250 int32_t dim_index);

12