Searched defs:index_shape (Results 1 – 17 of 17) sorted by relevance
37 const TensorShape& index_shape = ctx->InputShape(1); in Compile() local
41 BaseEmbeddingLookupOpModel(std::initializer_list<int> index_shape, in BaseEmbeddingLookupOpModel()90 HybridEmbeddingLookupOpModel(std::initializer_list<int> index_shape, in HybridEmbeddingLookupOpModel()
51 const TensorShape index_shape = ctx->InputShape("indices"); in Compile() local
106 const TensorShape index_shape = ctx->InputShape(2); in Compile() local
136 const Shape& index_shape = index_vector->shape(); in ExpandIndexVectorIntoOperandSpace() local
114 const Shape& index_shape = index_vector->shape(); in ExpandIndexVectorIntoOperandSpace() local
2539 for (const Shape& index_shape : start_index_shapes) { in InferDynamicSliceShape() local2646 for (const Shape& index_shape : start_index_shapes) { in InferDynamicUpdateSliceShape() local
1068 Shape index_shape = ShapeUtil::MakeShape(S32, {}); in TEST_F() local
2530 const Shape& index_shape = gather->operand(1)->shape(); in HandleGather() local
676 TensorShape index_shape({0, 3}); in ProcessBatch() local708 const TensorShape index_shape({batch.num_tasks(), 3}); in EmitIndexTensor() local
495 TfLiteIntArray* index_shape = TfLiteIntArrayCreate(2); in FastParseExampleLite() local
1021 auto index_shape = ShapeUtil::ChangeElementType(valid_slice->shape(), S32); in ExchangeHaloAndGetValidData() local
489 auto index_shape = ShapeUtil::ChangeElementType(shape, S32); in PadWithValue() local
1618 xla::Shape index_shape; in TEST_F() local
1979 ArrayRef<int64_t> index_shape = index_ty.getShape(); in Verify() local
3001 auto index_shape = index_type.getShape().vec(); in fold() local
4752 BaseEmbeddingLookupOpModel(std::initializer_list<int> index_shape, in BaseEmbeddingLookupOpModel()