| /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/compiler/xla/service/llvm_ir/ |
| D | ops.cc | 50 IrArray::Index dim_index({ir_builder->getInt64(i)}); in EmitDynamicUpdateSliceInPlaceImpl() local
51 TF_ASSIGN_OR_RETURN(start_index[i], start_indices_generator(dim_index)); in EmitDynamicUpdateSliceInPlaceImpl()
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | array_grad.py | 48 def _ConcatGradHelper(op, grad, start_value_index, end_value_index, dim_index): argument
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)
|
| D | nn_ops.py | 1660 def _swap_axis(logits, dim_index, last_index, name=None): argument
1665 math_ops.range(dim_index), [last_index],
1666 math_ops.range(dim_index + 1, last_index), [dim_index]
1854 def _move_dim_to_end(tensor, dim_index, rank): argument
1858 math_ops.range(dim_index),
1859 math_ops.range(dim_index + 1, rank), [dim_index]
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | shape_inference.cc | 660 int64 dim_index = dim_index_in; in ReplaceDim() local
661 if (dim_index < 0) { in ReplaceDim()
662 dim_index = s->dims_.size() + dim_index; in ReplaceDim()
664 if (!FastBoundsCheck(dim_index, s->dims_.size())) { in ReplaceDim()
671 dims[dim_index] = new_dim; in ReplaceDim()
|
| D | tensor.cc | 889 void PrintOneDim(int dim_index, const gtl::InlinedVector<int64, 4>& shape, in PrintOneDim() argument
893 int64 element_count = shape[dim_index]; in PrintOneDim()
895 if (dim_index == shape_size - 1) { in PrintOneDim()
911 PrintOneDim(dim_index + 1, shape, limit, shape_size, data, data_index, in PrintOneDim()
|
| D | common_shape_fns.cc | 1119 int end_value_index, int dim_index) { in ConcatShapeHelper() argument
1121 TF_RETURN_IF_ERROR(c->WithRank(c->input(dim_index), 0, &unused)); in ConcatShapeHelper()
1122 const Tensor* concat_dim_t = c->input_tensor(dim_index); in ConcatShapeHelper()
|
| D | shape_inference.h | 437 Status ReplaceDim(ShapeHandle s, int64 dim_index, DimensionHandle new_dim,
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | graph_properties.cc | 424 int dim_index; member
427 dim_index == other.dim_index; in operator ==()
433 return std::hash<const Node*>{}(dim.node) + dim.port_id + dim.dim_index; in operator ()()
|
| /external/tensorflow/tensorflow/c/eager/ |
| D | c_api.h | 124 int dim_index);
|
| D | c_api.cc | 159 int64_t TFE_TensorHandleDim(TFE_TensorHandle* h, int dim_index) { in TFE_TensorHandleDim() argument
160 return h->t.dim_size(dim_index); in TFE_TensorHandleDim()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | elemental_ir_emitter.cc | 1512 llvm_ir::IrArray::Index dim_index(1, ir_builder_->getInt64(i)); in MakeElementGenerator() local
1515 operand_to_generator.at(hlo->operand(1))(dim_index)); in MakeElementGenerator()
1556 llvm_ir::IrArray::Index dim_index(1, ir_builder_->getInt64(i)); in MakeElementGenerator() local
1558 operand_to_generator.at(start_hlo)(dim_index)); in MakeElementGenerator()
|
| /external/tensorflow/tensorflow/c/ |
| D | c_api.h | 266 TF_CAPI_EXPORT extern int64_t TF_Dim(const TF_Tensor* tensor, int dim_index);
|
| D | c_api.cc | 243 int64_t TF_Dim(const TF_Tensor* t, int dim_index) { in TF_Dim() argument
244 return static_cast<int64_t>(t->shape.dim_size(dim_index)); in TF_Dim()
|