Searched defs:dim_index (Results 1 – 11 of 11) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | reduction_ops_common.cc | 106 int dim_index = 0; in Simplify() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/ |
| D | ops.cc | 50 IrArray::Index dim_index({ir_builder->getInt64(i)}); in EmitDynamicUpdateSliceInPlaceImpl() local
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | nn_ops.py | 1660 def _swap_axis(logits, dim_index, last_index, name=None): argument
1854 def _move_dim_to_end(tensor, dim_index, rank): argument
|
| D | array_grad.py | 48 def _ConcatGradHelper(op, grad, start_value_index, end_value_index, dim_index): argument
|
| /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
1556 llvm_ir::IrArray::Index dim_index(1, ir_builder_->getInt64(i)); in MakeElementGenerator() local
|
| /external/tensorflow/tensorflow/c/eager/ |
| D | c_api.cc | 159 int64_t TFE_TensorHandleDim(TFE_TensorHandle* h, int dim_index) { in TFE_TensorHandleDim()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | shape_inference.cc | 660 int64 dim_index = dim_index_in; in ReplaceDim() local
|
| D | tensor.cc | 889 void PrintOneDim(int dim_index, const gtl::InlinedVector<int64, 4>& shape, in PrintOneDim()
|
| D | common_shape_fns.cc | 1119 int end_value_index, int dim_index) { in ConcatShapeHelper()
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | graph_properties.cc | 424 int dim_index; member
|
| /external/tensorflow/tensorflow/c/ |
| D | c_api.cc | 243 int64_t TF_Dim(const TF_Tensor* t, int dim_index) { in TF_Dim()
|