Searched defs:shape_spec (Results 1 – 5 of 5) sorted by relevance
106 const tstring& shape_spec = tensor_shapes_and_slices_ptr[i]; in SaveTensors() local195 const tstring& shape_spec = in RestoreTensor() local
118 const string& shape_spec = shape_and_slices_flat(i); in Compute() local
61 const StridedSliceShapeSpec& shape_spec, in EmitDynamicSlice()228 StridedSliceShapeSpec shape_spec; in Compile() local397 StridedSliceShapeSpec shape_spec; in CompileAsDynamicUpdateSlice() local
177 gtl::InlinedVector<int64, 4>* strides, StridedSliceShapeSpec* shape_spec) { in ValidateStridedSliceOp()425 StridedSliceShapeSpec* shape_spec) { in ValidateStridedSliceOp()
26 auto shape_spec = [&shape_spec_middle, shape_spec_end]( in TestGradAndIndicesErrorHandling() local