Searched defs:input_rank (Results 1 – 17 of 17) sorted by relevance
45 int input_rank = input_tensor_shape.dims(); in Compile() local
28 const int input_rank = input_tensor_shape.dims(); in SpaceToBatch() local
28 const int input_rank = input_tensor_shape.dims(); in BatchToSpace() local
77 const int input_rank = input_shape.dims(); in Compute() local
50 const int64 input_rank = input_shape_in.NumElements(); in Reshape() local
118 int input_rank = -1; in AnalyzeInputs() local
87 const int input_rank = NumDimensions(input); in Eval() local
488 int input_rank = c->Rank(c->input(0)); in TryToInferTensorOutputFromInputShapes() local516 int32 input_rank = c->Rank(c->input(0)); in TryToInferTensorOutputFromInputShapes() local
1261 const int32 input_rank = c->Rank(input); in __anondb9326b22302() local1663 const Tensor* paddings_t, int64 input_rank) { in MirrorPadKnown()1699 int64 input_rank = c->Value(pad_0); in __anondb9326b22d02() local1837 const int32 input_rank = c->Rank(input); in __anondb9326b23102() local
791 const int32 input_rank = c->Rank(input_shape); in ArgOpShape() local
1040 const int32 input_rank, in ReductionShapeHelper()1090 const int32 input_rank = c->Rank(input); in ReductionShape() local
857 int idx, int input_rank, DimensionHandle* out) { in MakeDimForScalarInputWithNegativeIndexing()
1271 int64 input_rank = Rank(input_shape); in AlignLayouts() local
628 auto input_rank = Size(scope, input_shape); in ReducedShapeHelper() local
2490 int input_rank, const int32* coords_data, in Gather()
3857 int input_rank, const int32* coords_data, in Gather()