Home
last modified time | relevance | path

Searched refs:TensorShapeDim (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/core/framework/
Dtensor_shape.h451 struct TensorShapeDim { struct
452 explicit TensorShapeDim(int64_t s) : size(s) {} in TensorShapeDim() argument
470 TensorShapeDim operator*() { return TensorShapeDim(shape_->dim_size(d_)); } argument
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/
Dcompile_mlir_util.cc350 for (tensorflow::TensorShapeDim dim : tensor_resource_shape.shape) { in RefineShapes()