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()
|