Searched defs:shape_spec (Results 1 – 3 of 3) sorted by relevance
106 const string& shape_spec = tensor_shapes_and_slices_ptr[i]; in SaveTensors() local195 const string& shape_spec = context->input(2).flat<string>()(restore_index); in RestoreTensor() local
117 const string& shape_spec = shape_and_slices_flat(i); in Compute() local
26 auto shape_spec = [&shape_spec_middle, shape_spec_end]( in TestGradAndIndicesErrorHandling() local