Searched refs:TensorShapeDim (Results 1 – 3 of 3) sorted by relevance
428 struct TensorShapeDim { struct429 explicit TensorShapeDim(int64 s) : size(s) {} in TensorShapeDim() function447 TensorShapeDim operator*() { return TensorShapeDim(shape_->dim_size(d_)); } argument
80 for (tensorflow::TensorShapeDim dimension : input_shape) { in Compile()
252 for (tensorflow::TensorShapeDim dim : tensor_resource_shape.shape) { in RefineShapes()