Searched refs:TensorShapeDim (Results 1 – 2 of 2) sorted by relevance
325 struct TensorShapeDim { struct326 explicit TensorShapeDim(int64 s) : size(s) {} in TensorShapeDim() argument344 TensorShapeDim operator*() { return TensorShapeDim(shape_->dim_size(d_)); } argument
74 for (tensorflow::TensorShapeDim dimension : input_shape) { in Compile()