Searched defs:flat_shape (Results 1 – 6 of 6) sorted by relevance
55 int flat_shape[] = {1, 1, 1, num_elements_}; in AddNumerator() local91 int flat_shape[] = {1, 1, 1, num_elements_}; in PopulateSubGraph() local
52 int64 flat_shape[] = {1}; in CreateStringVectorTensorHandle() local
66 auto flat_shape = shape_tensor.flat<int32>(); in FuzzImpl() local
71 auto flat_shape = ops::Const(s.WithOpName("flat_shape"), in CreateMiniGraph() local
792 TensorShape flat_shape = {static_cast<int64>(message_count), in Compute() local
895 const Shape& flat_shape = instr->shape(); in TryFlattenNestedTuples() local