Home
last modified time | relevance | path

Searched defs:dim_value (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/python/util/
Dpy_checkpoint_reader_wrapper.cc66 tensorflow::Safe_PyObjectPtr dim_value( in cast() local
/external/tensorflow/tensorflow/core/kernels/
Dreduce_join_op.cc57 int64_t dim_value = quotient % input_shape.dim_size(dim); in LinearSubIndexToFullIndex() local
/external/tensorflow/tensorflow/cc/gradients/
Darray_grad.cc638 Output dim_value = Slice(scope, casted_params_shape, {dim - 1}, {1}); in GetBatchIndices() local
/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/
Dlower_static_tensor_list.cc560 auto dim_value = (*it).getSExtValue(); in matchAndRewrite() local
/external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/
Dlegalize_common.cc1183 Value input_value, Value dim_value) { in convertExpandDimsOp()
/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
Dlegalize_tf.cc1167 Value dim_value = rewriter.create<tensor::DimOp>(loc, val, dim); in matchAndRewriteDynamicConv() local
3120 auto dim_value = rewriter.create<arith::IndexCastOp>( in matchAndRewrite() local