| /external/tensorflow/tensorflow/core/kernels/ |
| reduction_ops_common.cc | 106 int dim_index = 0; local
107 for (; dim_index < data.dims(); ++dim_index) {
108 if (data.dim_size(dim_index) != 1) break;
110 if (dim_index >= data.dims()) {
114 // Starting from the (dim_index)-th dimension, dimensions
124 reduce_first_axis_ = bitmap[dim_index];
125 data_reshape_.push_back(data.dim_size(dim_index));
126 ++dim_index;
127 for (; dim_index < data.dims(); ++dim_index)
[all...] |
| /external/tensorflow/tensorflow/c/eager/ |
| c_api_debug.cc | 130 int64 dim_index = padded_shape.layout().minor_to_major(i); local
131 dev_dims.push_back(padded_shape.dimensions(dim_index));
159 TFE_TensorDebugInfo* debug_info, int dim_index) {
160 return debug_info->dev_dims[dim_index];
|
| c_api.h | 171 int dim_index,
241 // Returns the number of elements in dimension `dim_index`.
243 // on host. The data contained in dimension `dim_index` on device
252 TFE_TensorDebugInfo* debug_info, int dim_index);
|
| c_api.cc | 466 int64_t TFE_TensorHandleDim(TFE_TensorHandle* h, int dim_index,
474 status->status = h->handle->Dim(dim_index, &result);
[all...] |
| /external/eigen/unsupported/Eigen/CXX11/src/Tensor/ |
| 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
[all...] |
| /external/tensorflow/tensorflow/core/framework/ |
| tensor.cc | [all...] |
| shape_inference.cc | 716 int64 dim_index = dim_index_in; local
717 if (dim_index < 0) {
718 dim_index = s->dims_.size() + dim_index;
720 if (!FastBoundsCheck(dim_index, s->dims_.size())) {
722 return errors::InvalidArgument("Out of range dim_index ", dim_index_in,
727 dims[dim_index] = new_dim;
[all...] |
| shape_inference.h | 451 // Returns in <out> the shape from replacing <s.dim[dim_index]> with
453 Status ReplaceDim(ShapeHandle s, int64 dim_index, DimensionHandle new_dim,
[all...] |
| common_shape_fns.cc | [all...] |
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| tensor_handle.cc | 202 Status TensorHandle::Dim(int dim_index, int64* dim) {
205 *dim = remote_shape_->dim_size(dim_index);
211 *dim = tensor_.dim_size(dim_index);
|
| tensor_handle.h | 95 Status Dim(int dim_index, int64* dim);
|
| /external/tensorflow/tensorflow/lite/experimental/c/ |
| c_api.h | 185 // Returns the length of the tensor in the "dim_index" dimension.
186 // REQUIRES: 0 <= dim_index < TFLiteTensorNumDims(tensor)
188 int32_t dim_index);
|
| c_api.cc | 172 int32_t TFL_TensorDim(const TFL_Tensor* tensor, int32_t dim_index) {
173 return tensor->dims->data[dim_index];
|
| /external/tensorflow/tensorflow/python/ops/ |
| array_grad.py | 47 def _ConcatGradHelper(op, grad, start_value_index, end_value_index, dim_index):
56 dim_index: An interger index of concat_dim or axis parameter in op.inputs.
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)
[all...] |
| nn_ops.py | [all...] |
| /external/skia/modules/skottie/src/ |
| SkottieAdapter.cpp | 306 const auto dim_index = SkTPin<size_t>(static_cast<size_t>(fDimensions), local
313 fBlur->setSigma({ sigma * kDimensionsMap[dim_index].x(),
314 sigma * kDimensionsMap[dim_index].y() });
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| graph_properties.cc | 915 int dim_index member in struct:tensorflow::grappler::SymbolicShapeRefiner::DimId
[all...] |
| /external/tensorflow/tensorflow/c/ |
| c_api.h | 265 // Return the length of the tensor in the "dim_index" dimension.
266 // REQUIRES: 0 <= dim_index < TF_NumDims(tensor)
267 TF_CAPI_EXPORT extern int64_t TF_Dim(const TF_Tensor* tensor, int dim_index);
[all...] |
| c_api.cc | 259 int64_t TF_Dim(const TF_Tensor* t, int dim_index) {
260 return static_cast<int64_t>(t->shape.dim_size(dim_index));
[all...] |
