| /external/tensorflow/tensorflow/core/kernels/ |
| D | reduction_ops_common.cc | 106 int dim_index = 0; in Simplify() local
107 for (; dim_index < data.dims(); ++dim_index) { in Simplify()
108 if (data.dim_size(dim_index) != 1) break; in Simplify()
110 if (dim_index >= data.dims()) { in Simplify()
124 reduce_first_axis_ = bitmap[dim_index]; in Simplify()
125 data_reshape_.push_back(data.dim_size(dim_index)); in Simplify()
126 ++dim_index; in Simplify()
127 for (; dim_index < data.dims(); ++dim_index) { in Simplify()
128 const auto size = data.dim_size(dim_index); in Simplify()
130 bitmap[dim_index] = bitmap[dim_index - 1]; in Simplify()
[all …]
|
| /external/eigen/unsupported/Eigen/CXX11/src/Tensor/ |
| D | TensorPadding.h | 205 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/tensorflow/tensorflow/c/eager/ |
| D | c_api_debug.cc | 130 int64 dim_index = padded_shape.layout().minor_to_major(i); in TFE_TensorHandleTensorDebugInfo() local
131 dev_dims.push_back(padded_shape.dimensions(dim_index)); in TFE_TensorHandleTensorDebugInfo()
159 TFE_TensorDebugInfo* debug_info, int dim_index) { in TFE_TensorDebugInfoOnDeviceDim() argument
160 return debug_info->dev_dims[dim_index]; in TFE_TensorDebugInfoOnDeviceDim()
|
| D | c_api.h | 171 int dim_index,
252 TFE_TensorDebugInfo* debug_info, int dim_index);
|
| D | c_api.cc | 466 int64_t TFE_TensorHandleDim(TFE_TensorHandle* h, int dim_index, in TFE_TensorHandleDim() argument
474 status->status = h->handle->Dim(dim_index, &result); in TFE_TensorHandleDim()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | tensor.cc | 961 void PrintOneDim(int dim_index, const gtl::InlinedVector<int64, 4>& shape, in PrintOneDim() argument
965 int64 element_count = shape[dim_index]; in PrintOneDim()
967 if (dim_index == shape_size - 1) { in PrintOneDim()
971 if (dim_index != 0 && i < element_count) { in PrintOneDim()
989 PrintOneDim(dim_index + 1, shape, limit, shape_size, data, data_index, in PrintOneDim()
999 void PrintDimSpacing(int dim_index, int num_dims, string* result) { in PrintDimSpacing() argument
1000 if (dim_index == num_dims - 1) { in PrintDimSpacing()
1004 for (int j = 0; j < num_dims - dim_index - 1; j++) { in PrintDimSpacing()
1007 for (int j = 0; j <= dim_index; j++) { in PrintDimSpacing()
1014 void PrintOneDimV2(int dim_index, const gtl::InlinedVector<int64, 4>& shape, in PrintOneDimV2() argument
[all …]
|
| D | shape_inference.cc | 716 int64 dim_index = dim_index_in; in ReplaceDim() local
717 if (dim_index < 0) { in ReplaceDim()
718 dim_index = s->dims_.size() + dim_index; in ReplaceDim()
720 if (!FastBoundsCheck(dim_index, s->dims_.size())) { in ReplaceDim()
727 dims[dim_index] = new_dim; in ReplaceDim()
|
| D | common_shape_fns.cc | 1212 int end_value_index, int dim_index) { in ConcatShapeHelper() argument
1214 TF_RETURN_IF_ERROR(c->WithRank(c->input(dim_index), 0, &unused)); in ConcatShapeHelper()
1215 const Tensor* concat_dim_t = c->input_tensor(dim_index); in ConcatShapeHelper()
|
| D | shape_inference.h | 453 Status ReplaceDim(ShapeHandle s, int64 dim_index, DimensionHandle new_dim,
|
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | tensor_handle.cc | 202 Status TensorHandle::Dim(int dim_index, int64* dim) { in Dim() argument
205 *dim = remote_shape_->dim_size(dim_index); in Dim()
211 *dim = tensor_.dim_size(dim_index); in Dim()
|
| D | tensor_handle.h | 95 Status Dim(int dim_index, int64* dim);
|
| /external/tensorflow/tensorflow/lite/experimental/c/ |
| D | c_api.cc | 172 int32_t TFL_TensorDim(const TFL_Tensor* tensor, int32_t dim_index) { in TFL_TensorDim() argument
173 return tensor->dims->data[dim_index]; in TFL_TensorDim()
|
| D | c_api.h | 188 int32_t dim_index);
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | array_grad.py | 47 def _ConcatGradHelper(op, grad, start_value_index, end_value_index, dim_index): argument
101 return grad + [None] if end_value_index <= dim_index else [None] + grad
103 concat_dim = op.inputs[dim_index]
207 if end_value_index <= dim_index else [None] + out_grads)
217 dim_index=0)
223 op, grad, start_value_index=0, end_value_index=-1, dim_index=-1)
|
| D | nn_ops.py | 2816 def _swap_axis(logits, dim_index, last_index, name=None): argument
2821 math_ops.range(dim_index), [last_index],
2822 math_ops.range(dim_index + 1, last_index), [dim_index]
3126 def _move_dim_to_end(tensor, dim_index, rank): argument
3130 math_ops.range(dim_index),
3131 math_ops.range(dim_index + 1, rank), [dim_index]
|
| /external/skia/modules/skottie/src/ |
| D | SkottieAdapter.cpp | 306 const auto dim_index = SkTPin<size_t>(static_cast<size_t>(fDimensions), in apply() local
313 fBlur->setSigma({ sigma * kDimensionsMap[dim_index].x(), in apply()
314 sigma * kDimensionsMap[dim_index].y() }); in apply()
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | graph_properties.cc | 915 int dim_index; member
918 dim_index == other.dim_index; in operator ==()
925 dim.dim_index; in operator ()()
|
| /external/tensorflow/tensorflow/c/ |
| D | c_api.h | 267 TF_CAPI_EXPORT extern int64_t TF_Dim(const TF_Tensor* tensor, int dim_index);
|
| D | c_api.cc | 259 int64_t TF_Dim(const TF_Tensor* t, int dim_index) { in TF_Dim() argument
260 return static_cast<int64_t>(t->shape.dim_size(dim_index)); in TF_Dim()
|