| /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/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/compiler/xla/service/llvm_ir/ |
| ops.cc | 50 IrArray::Index dim_index({ir_builder->getInt64(i)});
51 TF_ASSIGN_OR_RETURN(start_index[i], start_indices_generator(dim_index));
|
| /external/tensorflow/tensorflow/python/ops/ |
| array_grad.py | 48 def _ConcatGradHelper(op, grad, start_value_index, end_value_index, dim_index):
57 dim_index: An interger index of concat_dim or axis parameter in op.inputs.
102 return grad + [None] if end_value_index <= dim_index else [None] + grad
104 concat_dim = op.inputs[dim_index]
210 if end_value_index <= dim_index else [None] + out_grads)
220 dim_index=0)
226 op, grad, start_value_index=0, end_value_index=-1, dim_index=-1)
|
| nn_ops.py | [all...] |
| /external/tensorflow/tensorflow/core/framework/ |
| shape_inference.cc | 660 int64 dim_index = dim_index_in; local
661 if (dim_index < 0) {
662 dim_index = s->dims_.size() + dim_index;
664 if (!FastBoundsCheck(dim_index, s->dims_.size())) {
666 return errors::InvalidArgument("Out of range dim_index ", dim_index_in,
671 dims[dim_index] = new_dim;
[all...] |
| tensor.cc | 889 void PrintOneDim(int dim_index, const gtl::InlinedVector<int64, 4>& shape,
893 int64 element_count = shape[dim_index];
895 if (dim_index == shape_size - 1) {
911 PrintOneDim(dim_index + 1, shape, limit, shape_size, data, data_index,
[all...] |
| common_shape_fns.cc | [all...] |
| shape_inference.h | 435 // Returns in <out> the shape from replacing <s.dim[dim_index]> with
437 Status ReplaceDim(ShapeHandle s, int64 dim_index, DimensionHandle new_dim,
|
| /external/tensorflow/tensorflow/c/eager/ |
| c_api.h | 124 int dim_index);
|
| c_api.cc | 159 int64_t TFE_TensorHandleDim(TFE_TensorHandle* h, int dim_index) {
160 return h->t.dim_size(dim_index);
[all...] |
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| graph_properties.cc | 424 int dim_index; member in struct:tensorflow::grappler::SymbolicShapeRefiner::DimId
427 dim_index == other.dim_index;
433 return std::hash<const Node*>{}(dim.node) + dim.port_id + dim.dim_index;
[all...] |
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| elemental_ir_emitter.cc | [all...] |
| /external/tensorflow/tensorflow/c/ |
| c_api.h | 264 // Return the length of the tensor in the "dim_index" dimension.
265 // REQUIRES: 0 <= dim_index < TF_NumDims(tensor)
266 TF_CAPI_EXPORT extern int64_t TF_Dim(const TF_Tensor* tensor, int dim_index);
[all...] |
| c_api.cc | 243 int64_t TF_Dim(const TF_Tensor* t, int dim_index) {
244 return static_cast<int64_t>(t->shape.dim_size(dim_index));
[all...] |
